doubledsbv
commited on
Commit
•
0eb0e45
1
Parent(s):
bbbe239
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ generate_text = transformers.pipeline(
|
|
15 |
model=model, tokenizer=tokenizer,
|
16 |
return_full_text=True,
|
17 |
task='text-generation',
|
18 |
-
device="
|
19 |
)
|
20 |
|
21 |
@spaces.GPU
|
|
|
15 |
model=model, tokenizer=tokenizer,
|
16 |
return_full_text=True,
|
17 |
task='text-generation',
|
18 |
+
device="cuda",
|
19 |
)
|
20 |
|
21 |
@spaces.GPU
|