Spaces:

jboat
/

Text_Generation_NLLB_200

Sleeping

jboat commited on Jun 10

Commit

bc04d8f

•

1 Parent(s): 10882a6

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import pipeline
 app = FastAPI()
-pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 @app.get("/infer_t5")
 def t5(input):

 app = FastAPI()
+pipe_flan = pipeline("text2text-generation", model="facebook/nllb-200-distilled-1.3B")
 @app.get("/infer_t5")
 def t5(input):