Spaces:
Runtime error
Runtime error
update app
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_ckpt).to(device
|
|
| 16 |
|
| 17 |
class_names = ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise']
|
| 18 |
|
| 19 |
-
|
| 20 |
def inference(text: str) -> str:
|
| 21 |
inputs = tokenizer(text, return_tensors="pt")
|
| 22 |
|
|
|
|
| 16 |
|
| 17 |
class_names = ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise']
|
| 18 |
|
| 19 |
+
# main function
|
| 20 |
def inference(text: str) -> str:
|
| 21 |
inputs = tokenizer(text, return_tensors="pt")
|
| 22 |
|