Spaces:
Runtime error
Runtime error
thiruvanth
commited on
Commit
•
72b7afd
1
Parent(s):
292d488
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import torch
|
|
2 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
3 |
import torch
|
4 |
import numpy as np
|
|
|
5 |
|
6 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
7 |
|
@@ -11,6 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
|
|
11 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
|
12 |
model = model.to(device)
|
13 |
|
|
|
14 |
inputs = tokenizer(query, return_tensors='pt', truncation=True)
|
15 |
inputs = inputs.to(device)
|
16 |
outputs = model(**inputs)
|
@@ -45,7 +47,7 @@ def get_predictions(input_text: str) -> dict:
|
|
45 |
label2id = {k: float(v) for k, v in sorted(label2id.items(), key=lambda item: item[1].item(), reverse=True)}
|
46 |
return label2id
|
47 |
|
48 |
-
|
49 |
gr.Interface(
|
50 |
fn=get_predictions,
|
51 |
inputs=gr.components.Textbox(label='Input'),
|
|
|
2 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
3 |
import torch
|
4 |
import numpy as np
|
5 |
+
import gradio as gr
|
6 |
|
7 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
8 |
|
|
|
12 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
|
13 |
model = model.to(device)
|
14 |
|
15 |
+
query = 'I am not having a great day.'
|
16 |
inputs = tokenizer(query, return_tensors='pt', truncation=True)
|
17 |
inputs = inputs.to(device)
|
18 |
outputs = model(**inputs)
|
|
|
47 |
label2id = {k: float(v) for k, v in sorted(label2id.items(), key=lambda item: item[1].item(), reverse=True)}
|
48 |
return label2id
|
49 |
|
50 |
+
|
51 |
gr.Interface(
|
52 |
fn=get_predictions,
|
53 |
inputs=gr.components.Textbox(label='Input'),
|