Update app.py

app.py

@@ -109,7 +109,10 @@ def analyze_image(image: Image.Image, task: str = "<MORE_DETAILED_CAPTION>") ->
             image_size=RESIZE_DIM
         )
         
-        return result[task]
+        if result is None:
+            raise ValueError("Post-processing returned None. The model may not have generated a valid output for the given task.")
+
+        return result.get(task, "No caption generated.")
         
     except Exception as e:
         raise ValueError(f"Failed to analyze image: {e}")