Update app.py

app.py

@@ -1,8 +1,7 @@
 import os
 #import huggingface_hub
 #from huggingface_hub import snapshot_download
-#import open_oasis_master as oa 
-from open_oasis_master import generate
+import open_oasis_master as oa 
 os.system("huggingface-cli login --token $HF_TOKEN --add-to-git-credential")
 os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt --local-dir ./") # DiT checkpoint
 os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt --local-dir ./") 
@@ -10,7 +9,7 @@ os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt --local-dir ./
 
 with gr.Blocks() as app:
     vid = gr.Video()
-    app.load(generate.run_mod(),None,vid)
+    app.load(oa.generate.run_mod(),None,vid)
 app.launch()
 
 '''os.system("git clone https://github.com/etched-ai/open-oasis.git")