Update app.py

app.py

@@ -60,6 +60,9 @@ preprocess = transforms.Compose([
 
 # Function to perform predictions and show probabilities
 def predict(image):
+    if image is None:  # Check if no image was provided
+        return "Please insert photo"
+    
     # Apply preprocessing to the input image
     image = Image.fromarray(image).convert("RGB")
     input_tensor = preprocess(image).unsqueeze(0).to(device)
@@ -84,4 +87,4 @@ def predict(image):
 iface = gr.Interface(fn=predict, inputs=gr.Image(type="numpy"), outputs="text", live=True)
 
 # Launch the interface
-iface.launch()
+iface.launch()