Update app.py

app.py

@@ -46,16 +46,19 @@ def respond(
                 messages.append({"role": "user", "content": "Image uploaded", "image": image_b64})
 
     # Call Hugging Face model for response
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-
-        yield token
+    try:
+        for response in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if response.choices:
+                token = response.choices[0].delta.content
+                yield token
+    except Exception as e:
+        yield str(e)
 
 # Debugging print statements
 print("Starting Gradio interface setup...")