Update app.py

app.py

@@ -18,17 +18,14 @@ transform = transforms.Compose([
 # Function to classify posture images
 def classify_image(image):
     if image is None:
-        return "No image provided! Please upload or capture an image."
+        return "No image provided!"
     
     image = transform(image).unsqueeze(0)
     output = model(image)
     _, predicted = torch.max(output, 1)
-    return (
-        "Good Posture! Sit exactly like that for your Interview!" 
-        if predicted.item() == 0 
-        else "Bad Posture, you should think of sitting a little straighter or more in frame for your real interview."
-    )
+    return "Good Posture" if predicted.item() == 0 else "Bad Posture"
 
-# Set up Gradio interface
-iface = gr.Interface(fn=classify_image, inputs=gr.Image(type="pil"), outputs="text")
+# Set up Gradio interface with live webcam capture
+iface = gr.Interface(fn=classify_image, inputs=gr.Image(type="pil", live=True), outputs="text")
 iface.launch()
+