Hugging Face's logo

Spaces:
tuan243
/
ai_focus_new
Sleeping

App Files Community
tuan243 commited on
Commit
9bb19ba
·
verified ·
1 Parent(s): 6c73a7c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -18,7 +18,7 @@ if HF_TOKEN:
18
  login(HF_TOKEN)
19
 
20
  # Load AI Model
21
- ai_model = pipeline("text-generation", model="google/flan-t5-base")
22
 
23
  app = FastAPI()
24
 
 
18
  login(HF_TOKEN)
19
 
20
  # Load AI Model
21
+ ai_model = pipeline("text2text-generation", model="google/flan-t5-base")
22
 
23
  app = FastAPI()
24