Upload app.py with huggingface_hub

app.py

@@ -67,7 +67,7 @@ class DocumentClassifierCNN(nn.Module):
 
 def load_model():
     model = DocumentClassifierCNN()
-    ckpt = torch.load("best_model.pth", map_location="cpu", weights_only=True)
+    ckpt = torch.load("best_model.pth", map_location="cpu", weights_only=False)
     model.load_state_dict(ckpt["model_state_dict"])
     model.eval()
     return model
@@ -222,4 +222,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True)