macrdel commited on
Commit
bd4ab21
·
1 Parent(s): e42331d

update api.py

Browse files
Files changed (1) hide show
  1. app/api.py +3 -3
app/api.py CHANGED
@@ -1,12 +1,12 @@
1
- import config
2
- from src import pipeline_sentiment
3
 
4
  from fastapi import FastAPI
5
  from pydantic import BaseModel
6
  from transformers import pipeline
7
  import uvicorn
8
 
9
- sentiment_model = pipeline(config.sentiment_model)
10
  app = FastAPI()
11
 
12
  class YouTubeUrl(BaseModel):
 
1
+ import config.config as config
2
+ from app.src.src import pipeline_sentiment
3
 
4
  from fastapi import FastAPI
5
  from pydantic import BaseModel
6
  from transformers import pipeline
7
  import uvicorn
8
 
9
+ sentiment_model = pipeline(model=config.sentiment_model)
10
  app = FastAPI()
11
 
12
  class YouTubeUrl(BaseModel):