Update app.py

app.py

@@ -28,8 +28,9 @@ def chatbot(query, history, system_message, max_length, temperature, top_k, top_
         )
         # Decode generated text
         response = tokenizer.decode(final_outputs[0], skip_special_tokens=True)
+        response = response.split('"')[1]
             
-    return response.split('"')[1]
+    return response
 
 # Gradio ChatInterface
 demo = gr.ChatInterface(