phamngoctukts commited on
Commit
deacc32
·
verified ·
1 Parent(s): 190bb2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ tk = token = os.environ.get("HF_TOKEN")
19
  login(tk)
20
  model_id = "meta-llama/Llama-3.2-1B"
21
  text2text = pipeline(
22
- "text-generation",
23
  model=model_id,
24
  torch_dtype=torch.bfloat16,
25
  device_map="auto"
 
19
  login(tk)
20
  model_id = "meta-llama/Llama-3.2-1B"
21
  text2text = pipeline(
22
+ "text2text",
23
  model=model_id,
24
  torch_dtype=torch.bfloat16,
25
  device_map="auto"