Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def inference(input_text):
|
|
30 |
logits = model(**inputs).logits
|
31 |
|
32 |
predicted_class_id = logits.argmax().item()
|
33 |
-
output = model.config.
|
34 |
return output
|
35 |
|
36 |
demo = gr.Interface(
|
|
|
30 |
logits = model(**inputs).logits
|
31 |
|
32 |
predicted_class_id = logits.argmax().item()
|
33 |
+
output = model.config.id2label[predicted_class_id]
|
34 |
return output
|
35 |
|
36 |
demo = gr.Interface(
|