Update main.py
Browse files
main.py
CHANGED
|
@@ -55,7 +55,7 @@ async def lifespan(app: FastAPI):
|
|
| 55 |
# Use a pipeline as a high-level helper
|
| 56 |
from transformers import pipeline
|
| 57 |
global sentiment_task
|
| 58 |
-
pipe = pipeline("text-classification", model="SamLowe/roberta-base-go_emotions","SamLowe/roberta-base-go_emotions")
|
| 59 |
|
| 60 |
yield
|
| 61 |
# Clean up the model and release the resources
|
|
|
|
| 55 |
# Use a pipeline as a high-level helper
|
| 56 |
from transformers import pipeline
|
| 57 |
global sentiment_task
|
| 58 |
+
pipe = pipeline("text-classification", model="SamLowe/roberta-base-go_emotions", tokenizer="SamLowe/roberta-base-go_emotions")
|
| 59 |
|
| 60 |
yield
|
| 61 |
# Clean up the model and release the resources
|