Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+ import gradio as gr
 import torch
 import torchvision.transforms as transforms
 import torchvision.models as models
@@ -17,11 +17,14 @@ transform = transforms.Compose([
 
 # Function to classify posture images
 def classify_image(image):
+    if image is None:
+        return "No image provided!"
+    
     image = transform(image).unsqueeze(0)
     output = model(image)
     _, predicted = torch.max(output, 1)
     return "Good Posture" if predicted.item() == 0 else "Bad Posture"
 
 # Set up Gradio interface for webcam or image upload
-iface = gr.Interface(fn=classify_image, inputs=gr.Image(type="pil"), outputs="text")
+iface = gr.Interface(fn=classify_image, inputs=gr.Image(type="pil", source="upload"), outputs="text")
 iface.launch()