Update app.py

app.py

@@ -16,19 +16,15 @@ import random
 from huggingface_hub import login, hf_hub_download
 import spaces
 
-#gradio.helpers.CACHED_FOLDER = '/data/cache'
-
-# SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
-
-# HF_API_KEY = os.getenv('HF_API_KEY', '')
-# login(token=HF_API_KEY)
-
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     # "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
     "vdo/stable-video-diffusion-img2vid-xt-1-1",
     torch_dtype=torch.float16,
     variant="fp16"
 )
+
+pipe.save_pretrained("model", variant="fp16")
+
 pipe.to("cuda")
 pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 #pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
@@ -37,7 +33,6 @@ max_64_bit_int = 2**63 - 1
 
 @spaces.GPU(enable_queue=True)
 def generate_video(
-    secret_token: str,
     image: Image,
     seed: int,
     motion_bucket_id: int = 127,
@@ -48,11 +43,6 @@ def generate_video(
     device: str = "cuda",
     output_folder: str = "outputs",
 ):
-    # if secret_token != SECRET_TOKEN:
-    #     raise gr.Error(
-    #         f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
-
-
     # note julian: normally we should resize input images, but normally they are already in 1024x576, so..
 
     # also, I would like to experiment with vertical videos, and 1024x512 videos
@@ -116,11 +106,6 @@ def resize_image(image, output_size=(1024, 576)):
     return cropped_image
 
 with gr.Blocks() as demo:
-    # secret_token = gr.Text(
-    #     label='Secret Token',
-    #     max_lines=1,
-    #     placeholder='Enter your secret token')
-
     image = gr.Image(label="Upload your image", type="pil")
     generate_btn = gr.Button("Generate")
     base64_out = gr.Textbox(label="Base64 Video")
@@ -130,7 +115,6 @@ with gr.Blocks() as demo:
        
     generate_btn.click(
         fn=generate_video,
-        # inputs=[secret_token, image, seed, motion_bucket_id, fps_id],
         inputs=[image, seed, motion_bucket_id, fps_id],
         outputs=base64_out,
         api_name="run"