Update space

app.py

@@ -16,31 +16,21 @@ def respond(
     temperature,
     top_p,
 ):
-    # Format conversation history into a text prompt
-    prompt = f"{system_message}\n\n"
+    # Simpler prompt format
+    prompt = message
     
-    for user_msg, assistant_msg in history:
-        prompt += f"User: {user_msg}\n"
-        if assistant_msg:
-            prompt += f"Assistant: {assistant_msg}\n"
-    
-    prompt += f"User: {message}\nAssistant:"
-
     response = ""
     try:
-        # Use text_generation instead of chat_completion
-        for chunk in client.text_generation(
+        # Basic text generation without streaming first
+        response = client.text_generation(
             prompt,
             max_new_tokens=max_tokens,
             temperature=temperature,
             top_p=top_p,
-            stream=True
-        ):
-            if chunk:
-                response += chunk
-                yield response
+        )
+        return response
     except Exception as e:
-        yield f"Error: {str(e)}"
+        return f"Error: {str(e)}"
 
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface