Spaces:

pivovalera2012
/

dr_House

Runtime error

pivovalera2012 commited on Mar 2, 2024

Commit

bb09273

verified ·

1 Parent(s): 33d8076

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 def generate_text(prompt):
-    encoding = tokenizer(prompt, return_tensors="pt").to(device)
     with torch.no_grad():
         outputs = model_trained.generate(
             input_ids = encoding.input_ids,

 def generate_text(prompt):
+    encoding = tokenizer(prompt, return_tensors="pt")
     with torch.no_grad():
         outputs = model_trained.generate(
             input_ids = encoding.input_ids,