Spaces:
Runtime error
Runtime error
Commit
·
764f570
1
Parent(s):
ce18e89
Update app.py
Browse files
app.py
CHANGED
@@ -57,8 +57,7 @@ def url_changes(url, pages_to_visit, urls_to_scrape, repo_id):
|
|
57 |
model = AutoModelForCausalLM.from_pretrained(
|
58 |
MODEL,
|
59 |
torch_dtype="auto",
|
60 |
-
|
61 |
-
).to(device=f"cuda", non_blocking=True)
|
62 |
model.eval()
|
63 |
pipe = pipeline(
|
64 |
'text-generation',
|
|
|
57 |
model = AutoModelForCausalLM.from_pretrained(
|
58 |
MODEL,
|
59 |
torch_dtype="auto",
|
60 |
+
)
|
|
|
61 |
model.eval()
|
62 |
pipe = pipeline(
|
63 |
'text-generation',
|