Update app.py

app.py

@@ -9,13 +9,16 @@ model = BlipForConditionalGeneration.from_pretrained("quadranttechnologies/Dilee
 
 # Define a function to generate captions for the uploaded image
 def generate_caption(image):
-    # Convert the image into the required format for the model
-    inputs = processor(image, return_tensors="pt")
+    try:
+        # Convert the image into the required format for the model
+        inputs = processor(image, return_tensors="pt")
 
-    # Generate caption
-    outputs = model.generate(**inputs)
-    caption = processor.decode(outputs[0], skip_special_tokens=True)
-    return caption
+        # Generate caption
+        outputs = model.generate(**inputs)
+        caption = processor.decode(outputs[0], skip_special_tokens=True)
+        return caption
+    except Exception as e:
+        return f"Error generating caption: {e}"
 
 # Set up Gradio interface for image upload and caption generation
 interface = gr.Interface(
@@ -28,4 +31,5 @@ interface = gr.Interface(
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)  # Set share=True to enable public link if needed
+