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',
|