Spaces:

SAHILVAGHASIYA
/

text2textsentence

Runtime error

SAHILVAGHASIYA commited on 12 days ago

Commit

acb25e0

•

1 Parent(s): 28a215e

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ app=FastAPI()
 import transformers
 import torch
-model_id = "meta-llama/Meta-Llama-3-8B"
 pipeline = transformers.pipeline(
     "text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"

 import transformers
 import torch
+model_id = "google/flan-t5-base"
 pipeline = transformers.pipeline(
     "text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"