Upload app.py

app.py

@@ -6,12 +6,15 @@ os.system("pip3 install torch torchvision torchaudio --index-url https://downloa
 import gradio as gr
 import torch
 import subprocess
-
+from main import create_original
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+secret_api = os.getenv("YOUR_GEMINI_API")
+
 def infer():
-    os.systtem("call make_original_video.bat")
+    title = None
+    create_original(api_key=secret_api, des=title)
     file_path = f"result/output.mp4"
     return file_path
 
@@ -30,6 +33,7 @@ with gr.Blocks() as demo:
         gr.Markdown(f"""
         #      Short Video Generator
         -- Click run to generate video --
+                    
         Currently running on {power_device}.
         """)
 
@@ -37,6 +41,7 @@ with gr.Blocks() as demo:
             run_button = gr.Button("Run", scale=0)
             change_track_button = gr.Button("Change Track", scale=0)
 
+        # title = gr.Textbox()
         result = gr.Video()
         result_changed = gr.Video()