Update app.py

app.py

@@ -131,8 +131,8 @@ chain = prompt | chat_model
 def handle_input(image, text, session_id):
     if image:
         # Handle image input
-        image_desc = category(image)
-        return image_desc
+        response = category(image)
+        
     else:
         # Handle text input
         if text=='':