Spaces:
Running
Running
AyoubChLin
commited on
Commit
•
aa204f5
1
Parent(s):
31001c6
Update app.py
Browse files
app.py
CHANGED
@@ -16,8 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3.5-mini-instruct")
|
|
16 |
pipe = pipeline(
|
17 |
"text-generation",
|
18 |
model=model,
|
19 |
-
tokenizer=tokenizer
|
20 |
-
device="cpu"
|
21 |
)
|
22 |
|
23 |
# Generation arguments
|
|
|
16 |
pipe = pipeline(
|
17 |
"text-generation",
|
18 |
model=model,
|
19 |
+
tokenizer=tokenizer
|
|
|
20 |
)
|
21 |
|
22 |
# Generation arguments
|