Pengi5659 commited on
Commit
6216147
·
verified ·
1 Parent(s): 27c5834

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def classify_image(image):
23
  image = transform(image).unsqueeze(0)
24
  output = model(image)
25
  _, predicted = torch.max(output, 1)
26
- 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 for your real interview."
27
 
28
  # Set up Gradio interface
29
  iface = gr.Interface(fn=classify_image, inputs=gr.Image(type="pil"), outputs="text")
 
23
  image = transform(image).unsqueeze(0)
24
  output = model(image)
25
  _, predicted = torch.max(output, 1)
26
+ 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."
27
 
28
  # Set up Gradio interface
29
  iface = gr.Interface(fn=classify_image, inputs=gr.Image(type="pil"), outputs="text")