saideep-arikontham commited on
Commit
8b9a20c
1 Parent(s): 73ae739

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ tokenizer = AutoTokenizer.from_pretrained(adapter_model)
18
 
19
  def greet(text):
20
 
21
- model.to('cuda')
22
- inputs = tokenizer.encode(text, return_tensors="pt").to("cuda")
23
  # compute logits
24
  logits = model(inputs).logits
25
  # convert logits to label
 
18
 
19
  def greet(text):
20
 
21
+ model.to('cpu')
22
+ inputs = tokenizer.encode(text, return_tensors="pt").to("cpu")
23
  # compute logits
24
  logits = model(inputs).logits
25
  # convert logits to label