Upload app.py

app.py

@@ -142,7 +142,7 @@ if __name__ == "__main__":
     deberta_tokenizer = AutoTokenizer.from_pretrained("microsoft/deberta-large", model_max_length=256)
 
     model = Custom_bert("microsoft/deberta-large")
-    model.load_state_dict(torch.load(deberta_loc))
+    model.load_state_dict(torch.load(deberta_loc, map_location=torch.device(device)))
     model.eval().to(device)