Spaces:

sagarkariya
/

ProjectTestReporting

Sleeping

sagarkariya commited on 25 days ago

Commit

8102ed4

verified ·

1 Parent(s): d0174ef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ model.to(device)
 # 🔍 Function to Generate Responses
 def generate_response(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
     output = model.generate(**inputs, max_length=300)
     return tokenizer.decode(output[0], skip_special_tokens=True)

 # 🔍 Function to Generate Responses
 def generate_response(prompt):
+    inputs = tokenizer(prompt, return_tensors="pt").to("cpu")
     output = model.generate(**inputs, max_length=300)
     return tokenizer.decode(output[0], skip_special_tokens=True)