Update main.py

main.py

@@ -38,8 +38,14 @@ async def save_upload_file(upload_file: UploadFile) -> str:
 async def get_caption(image: UploadFile = File(...), context: str = None):
     # Save the uploaded image to a temporary file
     temp_file_path = await save_upload_file(image)
-    # Pass the temporary file path to the client for prediction
+    
+    # Check if additional context is provided and not None
+    if context is not None:
+        context = context.strip()
+    
+    # Pass the temporary file path and context to the Gradio client for prediction
     result = client.predict(temp_file_path, context, api_name="/get_caption")
+    
     return {"caption": result}
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")