Update app.py

app.py

@@ -19,7 +19,8 @@ llm = Llama(
 history = []
 
 system_message = """
-You are a helpful BIRD ASSISTANT, AND YOU KNOW A LOT ABOUT BIRDS"""
+You are a helpful, Bird researcher. you know alot about birds
+"""
 # The rest of the system message
 
 # Load the Vision Model
@@ -72,7 +73,7 @@ def predict(img):
         message += f"- {bird}: {prob:.2%}\n"
     
     # Generate responses using LLM
-    responses = generate_text(message, history)
+    responses = list(generate_text(message, history))
     
     return {"bird_predictions": bird_predictions, "llm_responses": responses}
 
@@ -84,11 +85,11 @@ enable_queue = True
 
 gr.Interface(
     fn=predict,
-    inputs=gr.inputs.Image(shape=(512, 512)),
-    outputs={
-        "bird_predictions": gr.outputs.Data(type="dictionary"),
-        "llm_responses": gr.outputs.Textbox(),
-    },
+    inputs=gr.inputs.Image(),
+    outputs=[
+        gr.outputs.Data(type="dictionary", label="Bird Predictions"),
+        gr.outputs.Textbox(label="LLM Responses")
+    ],
     title=title,
     description=description,
     examples=examples,