Update app.py

app.py

@@ -14,6 +14,7 @@ def image_to_base64(image):
     img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
     return img_str
 
+# Function to interact with LLAVA model
 # Function to interact with LLAVA model
 # Function to interact with LLAVA model
 def respond(
@@ -37,8 +38,8 @@ def respond(
         messages.append({"role": "user", "content": message})
 
         if image:
-            # Convert image to base64
-            image_b64 = image_to_base64(image)
+            # Handle single image input
+            image_b64 = image_to_base64(image)  # Convert image to base64
             messages.append({"role": "user", "content": "Image uploaded", "image": image_b64})
 
         # Call Hugging Face model for response
@@ -59,6 +60,7 @@ def respond(
         print(f"Error in respond function: {str(e)}")
         yield f"Error occurred: {str(e)}"
 
+
 # Debugging print statements
 print("Starting Gradio interface setup...")
 try: