first commit

app.py

@@ -16,7 +16,6 @@ pipe = pipeline(
     use_auth_token=hf_token
 )
 @spaces.GPU
-
 def generate_response(user_text, user_image, history):
     messages = [
         {
@@ -31,22 +30,15 @@ def generate_response(user_text, user_image, history):
         user_content.append({"type": "text", "text": user_text})
     messages.append({"role": "user", "content": user_content})
     
+    # Call the pipeline with the provided messages.
     output = pipe(text=messages, max_new_tokens=200)
-    response = output[0][0]["generated_text"][-1]["content"]
+    
+    # Attempt to extract the generated content using the expected structure.
+    try:
+        response = output[0][0]["generated_text"][-1]["content"]
+    except (KeyError, IndexError, TypeError):
+        # Fallback: return the raw output as a string.
+        response = str(output)
+    
     history.append((user_text, response))
-    return history, history
-
-with gr.Blocks() as demo:
-    gr.Markdown("# Gemma 3 Chat Interface")
-    gr.Markdown(
-        "This interface lets you chat with the Gemma 3 model. "
-        "You can type a message and optionally attach an image."
-    )
-    chatbot = gr.Chatbot(type="messages")
-    with gr.Row():
-        txt = gr.Textbox(show_label=False, placeholder="Type your message here...", container=False)
-        img = gr.Image(type="pil", label="Attach an image (optional)")
-    state = gr.State([])
-
-    txt.submit(generate_response, inputs=[txt, img, state], outputs=[chatbot, state])
-    demo.launch()
+    return history, history