Update app.py

app.py

@@ -2,7 +2,8 @@ import os
 import gradio as gr
 #import huggingface_hub
 #from huggingface_hub import snapshot_download
-import open_oasis_master as oa 
+import open_oasis_master
+from open_oasis_master import generate
 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 +11,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(oa.generate.run_mod(),None,vid)
+    app.load(generate.run_mod(),None,vid)
 app.launch()
 
 '''os.system("git clone https://github.com/etched-ai/open-oasis.git")