mohsinabbas1984 commited on
Commit
e558606
1 Parent(s): f360131

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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