Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def predict_intent(text):
|
|
20 |
logits = outputs.logits
|
21 |
predicted_class = torch.argmax(F.softmax(logits, dim=1)).item()
|
22 |
intent_name = [k for k, v in intent_labels.items() if v == predicted_class][0]
|
23 |
-
return
|
24 |
|
25 |
# Gradio Interface
|
26 |
demo = gr.Interface(
|
|
|
20 |
logits = outputs.logits
|
21 |
predicted_class = torch.argmax(F.softmax(logits, dim=1)).item()
|
22 |
intent_name = [k for k, v in intent_labels.items() if v == predicted_class][0]
|
23 |
+
return intent_name
|
24 |
|
25 |
# Gradio Interface
|
26 |
demo = gr.Interface(
|