Spaces:

brianronan
/

netron-docker

Sleeping

brian-xetdata commited on Aug 16, 2024

Commit

0ac63a7

1 Parent(s): 6d5ce82

Defensive coding

Files changed (1) hide show

proxy.py CHANGED Viewed

@@ -42,10 +42,16 @@ from huggingface_hub import hf_hub_url
 @app.get("/")
 def chooser():
     file_url = hf_hub_url(request.args.get("repo_id"), request.args.get("filename"))
     if file_url is None:
-        return "Invalid URL"
     return """
 <html>

 @app.get("/")
 def chooser():
+    repo_id = request.args.get("repo_id")
+    filename = request.args.get("filename")
+    if repo_id is None or filename is None:
+        return ""
     file_url = hf_hub_url(request.args.get("repo_id"), request.args.get("filename"))
     if file_url is None:
+        return ""
     return """
 <html>