Spaces:

saideep-arikontham
/

biden_stance_detector

Sleeping

saideep-arikontham commited on Apr 24

Commit

5f3e327

•

1 Parent(s): fad4bb8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,8 +21,8 @@ tokenizer = AutoTokenizer.from_pretrained(adapter_model)
 def greet(text):
-    model.to('mps')
-    inputs = tokenizer.encode(text, return_tensors="pt").to("mps")
     # compute logits
     logits = model(inputs).logits
     # convert logits to label

 def greet(text):
+    model.to('cpu')
+    inputs = tokenizer.encode(text, return_tensors="pt").to("cpu")
     # compute logits
     logits = model(inputs).logits
     # convert logits to label