Update app.py

app.py

@@ -16,7 +16,7 @@ def Multi_Label_Classification_of_Pubmed_Articles(model_input: str) -> Dict[str,
     
     # Encoding input data
     encodings = tokenizer.batch_encode_plus(model_input,max_length=1024,padding=True,truncation=True) # tokenizer's encoding method
-    outs = model(torch.tensor(encodings['input_ids']).cuda(), token_type_ids=None, attention_mask=torch.tensor(encodings['attention_mask']).cuda())
+    outs = model(torch.tensor(encodings['input_ids']), token_type_ids=None, attention_mask=torch.tensor(encodings['attention_mask']))
     b_logit_pred = outs[0]
     pred_label = torch.sigmoid(b_logit_pred)
     
@@ -74,6 +74,7 @@ article = (
 )
 
 
+
 app = gr.Interface(
     Multi_Label_Classification_of_Pubmed_Articles, 
 	inputs=model_input,