Update app.py

app.py

@@ -32,7 +32,7 @@ tag_to_idx = {'cs': 0,
 
 tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
 
-model = AutoModel.from_pretrained('./model', device_map='cpu')
+model = AutoModel.from_pretrained('./model', device_map='CPU')
 
 
 st.markdown("### Угадыватель")
@@ -46,8 +46,8 @@ ans = None
 
 if st.button('Предположить'):
     inputs = tokenizer(title + ' : ' + abstract, return_tensors='pt')
-    inputs['input_ids'] = inputs['input_ids'].to("cuda")
-    inputs['attention_mask'] = inputs['attention_mask'].to("cuda")
+    inputs['input_ids'] = inputs['input_ids']
+    inputs['attention_mask'] = inputs['attention_mask']
 
     with torch.no_grad():
         logits = model(**inputs).logits
@@ -58,8 +58,8 @@ if st.button('Предположить'):
 if st.button("Посмотреть топ"):
     if not logits:
         inputs = tokenizer(title + ' : ' + abstract, return_tensors='pt')
-        inputs['input_ids'] = inputs['input_ids'].to("cuda")
-        inputs['attention_mask'] = inputs['attention_mask'].to("cuda")
+        inputs['input_ids'] = inputs['input_ids']
+        inputs['attention_mask'] = inputs['attention_mask']
 
         with torch.no_grad():
             logits = model(**inputs).logits