Upload folder using huggingface_hub

demo.py

@@ -89,8 +89,10 @@ def predict(inp):
 
 print(f"=> Model (tpmlp_{size}) loaded in {time.time()- start:.2f} sec(s).")
 
-if not os.path.isdir("../example-imgs"):
-    os.mkdir("../example-imgs")
+base = ".." if args.local else "."
+
+if not os.path.isdir(f"{base}/example-imgs"):
+    os.mkdir(f"{base}/example-imgs")
 
 print("=> Loading examples.")
 indices = [
@@ -108,7 +110,7 @@ examples = []; idx = 0
 start = time.time()
 for data in ds:
     if idx == indices:
-        data['image'].save(f"../example-imgs/{idx}.png")
+        data['image'].save(f"{base}/example-imgs/{idx}.png")
     idx += 1
     if idx == max(indices):
         break
@@ -137,9 +139,9 @@ with gr.Blocks(theme=gr.themes.Monochrome(font=[gr.themes.GoogleFont("DM Sans"),
         gr.Button("Predict").click(fn=predict, inputs=input_image, outputs=[softmax, grad_cam])
         gr.ClearButton(input_image)
     with gr.Row():
-        gr.Examples([f"../example-imgs/{idx}.png" for idx in indices], inputs=input_image, outputs=[softmax, grad_cam], fn=predict, run_on_click=True)
+        gr.Examples([f"{base}/example-imgs/{idx}.png" for idx in indices], inputs=input_image, outputs=[softmax, grad_cam], fn=predict, run_on_click=True)
             
 demo.launch(
-    share=False, debug=False, allowed_paths=["../example-imgs"], server_name="0.0.0.0", # ssl_verify=False,
+    share=False, debug=False, allowed_paths=[f"{base}/example-imgs"], server_name="0.0.0.0", # ssl_verify=False,
     server_port=8000, # ssl_certfile="/workspace/openssl/cert.pem", ssl_keyfile="/workspace/openssl/key.pem"
 )