fix 6

app.py

@@ -26,6 +26,33 @@ output = query({
 })
 '''
 
+def generate_text(
+    prompt,
+    system_message,
+    max_tokens,
+    temperature,
+    top_p,
+):
+    try:
+        logger.info(f"Attempting to generate text for prompt: {prompt[:50]}...")
+        
+        response = client.text_generation(
+            prompt,
+            max_new_tokens=max_tokens,
+            temperature=temperature,
+            top_k=50,
+            top_p=top_p,
+            do_sample=True
+        )
+        
+        logger.info(f"Generated text: {response[:100]}...")
+        return response
+    except Exception as e:
+        logger.error(f"Error in generate_text: {type(e).__name__}: {str(e)}")
+        return f"An error occurred: {type(e).__name__}: {str(e)}"
+
+
+
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -63,7 +90,8 @@ def respond(
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
 demo = gr.ChatInterface(
-    respond,
+    #respond,
+    generate_text,
     additional_inputs=[
         gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),