Update app.py
Browse files
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")
|