Spaces:
Runtime error
Runtime error
phamngoctukts
commited on
Update app.py
Browse files
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 |
-
"
|
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"
|