Upload folder using huggingface_hub

demo.py

@@ -89,10 +89,7 @@ def predict(inp):
 
 print(f"=> Model (tpmlp_{size}) loaded in {time.time()- start:.2f} sec(s).")
 
-base = ".." if args.local else "."
-
-if not os.path.isdir(f"{base}/example-imgs"):
-    os.mkdir(f"{base}/example-imgs")
+base = "../example-imgs" if args.local else "."
 
 print("=> Loading examples.")
 indices = [
@@ -110,7 +107,7 @@ examples = []; idx = 0
 start = time.time()
 for data in ds:
     if idx == indices:
-        data['image'].save(f"{base}/example-imgs/{idx}.png")
+        data['image'].save(f"{base}/{idx}.png")
     idx += 1
     if idx == max(indices):
         break
@@ -139,9 +136,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"{base}/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}/{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=[f"{base}/example-imgs"], server_name="0.0.0.0", # ssl_verify=False,
+    share=False, debug=False, allowed_paths=[f"{base}"], server_name="0.0.0.0", # ssl_verify=False,
     server_port=8000, # ssl_certfile="/workspace/openssl/cert.pem", ssl_keyfile="/workspace/openssl/key.pem"
 )