Update app.py

app.py

@@ -215,7 +215,7 @@ def server(input: Inputs, output: Outputs, session: Session):
         image, prob, prediction = get_predictions()
 
         # Check if there are no predictions (i.e., no uploaded image)
-        if processed_image is None or prob is None or prediction is None:
+        if image is None or prob is None or prediction is None:
             # Return a placeholder plot or message
             fig, ax = plt.subplots()
             ax.text(0.5, 0.5, "Upload an image to see predictions", fontsize=12, ha="center")