Update app.py

app.py

@@ -5,10 +5,13 @@ import json
 from huggingface_hub import hf_hub_download
 
 REPO = "gabrielnkl/model-fraud-detect"
-MODEL = "model.pt"
+MODEL = "fraud_detect_model.pth"
 
+# download once + cache
 model_path = hf_hub_download(repo_id=REPO, filename=MODEL)
-model = torch.jit.load(model_path)
+
+# load model
+model = torch.load(model_path, map_location=torch.device("cpu"))
 model.eval()
 
 TYPE_MAP = {
@@ -51,5 +54,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
-
+    iface.launch(server_name="0.0.0.0", server_port=7860)