Update app.py

app.py

@@ -1,11 +1,12 @@
 import os
-import gradio as gr
-'''from huggingface_hub import hf_hub_download
-
+import huggingface_hub
 os.system("git clone https://github.com/etched-ai/open-oasis.git")
 os.system("cd open-oasis")
 os.system("pip install -r open-oasis/requirements.txt")
-oas = hf_hub_download(repo_id="Etched/oasis-500m", token=None, filename="oasis500m.pt")
-vit = hf_hub_download(repo_id="Etched/oasis-500m", token=None, filename="vit-l-20.pt")
-print(oas,vit)'''
-gr.load("models/Etched/oasis-500m").launch()
+os.system("huggingface-cli login")
+os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt") # DiT checkpoint
+os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt") 
+#from huggingface_hub import hf_hub_download
+#oas = hf_hub_download(repo_id="Etched/oasis-500m", filename="oasis500m.pt")
+#vit = hf_hub_download(repo_id="Etched/oasis-500m", filename="vit-l-20.pt")
+#print(oas,vit)