ai_focus_new

Sleeping

tuan243 commited on Mar 31

Commit

35c0e49

verified ·

1 Parent(s): 585ae8c

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -20,7 +20,7 @@ if HF_TOKEN:
 # Load AI Model
 tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base",token=HF_TOKEN)
-ai_model = AutoModelForCausalLM.from_pretrained("google/flan-t5-base",token=HF_TOKEN)
 app = FastAPI()

 # Load AI Model
 tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base",token=HF_TOKEN)
+ai_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base",token=HF_TOKEN)
 app = FastAPI()