Spaces:

saideep-arikontham
/

trump_stance_detector

Sleeping

saideep-arikontham commited on Apr 20

Commit

73ae739

•

1 Parent(s): 4277173

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,8 +18,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('cuda')
+    inputs = tokenizer.encode(text, return_tensors="pt").to("cuda")
     # compute logits
     logits = model(inputs).logits
     # convert logits to label