Update app.py

app.py

@@ -12,13 +12,13 @@ def generate_caption(image):
     
     # Preprocess the image and generate a detailed caption
     inputs = processor(images=image, return_tensors="pt")
-    out = model.generate(**inputs, num_beams=5, max_length=500, early_stopping=True)
+    out = model.generate(**inputs, num_beams=3, max_length=500, early_stopping=True)
     caption = processor.decode(out[0], skip_special_tokens=True)
     
     end_time = time.time()  # End time
     processing_time = end_time - start_time  # Calculate processing time
     
-    return caption, f"Time taken: {processing_time:.2f} seconds"
+    return caption, f"Time taken: {processing_time:.1f} seconds"
 
 # Function to handle cancellation (clear image and output)
 def cancel_action():