sonisatish119 commited on
Commit
923d55b
·
verified ·
1 Parent(s): b5e004b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 f"Intent: {intent_name}
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(