hpoghos commited on
Commit
2362b20
·
verified ·
1 Parent(s): c886731

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import torch
2
- torch.jit.script = lambda f: f
3
  # General
4
  import os
5
  from os.path import join as opj
6
  import argparse
7
  import datetime
8
  from pathlib import Path
9
- import spaces
10
  import gradio as gr
11
  import tempfile
12
  import yaml
@@ -59,7 +59,7 @@ msxl_model = init_v2v_model(cfg_v2v, devices[3])
59
  # -------------------------
60
  # ----- Functionality -----
61
  # -------------------------
62
- @spaces.GPU(duration=120)
63
  def generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance, where_to_log=result_fol):
64
  now = datetime.datetime.now()
65
  name = prompt[:100].replace(" ", "_") + "_" + str(now.time()).replace(":", "_").replace(".", "_")
@@ -89,7 +89,7 @@ def generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, se
89
  video_path = opj(where_to_log, name+".mp4")
90
  return video_path
91
 
92
- @spaces.GPU(duration=400)
93
  def enhance(prompt, input_to_enhance, num_frames=None, image=None, model_name_stage1=None, model_name_stage2=None, seed=33, t=50, image_guidance=9.5, result_fol=result_fol):
94
  if input_to_enhance is None:
95
  input_to_enhance = generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance)
@@ -176,7 +176,7 @@ with gr.Blocks() as demo:
176
  with gr.Row():
177
  num_frames = gr.Dropdown(["24 - frames", "32 - frames", "40 - frames", "48 - frames", "56 - frames", "80 - recommended to run on local GPUs", "240 - recommended to run on local GPUs", "600 - recommended to run on local GPUs", "1200 - recommended to run on local GPUs", "10000 - recommended to run on local GPUs"], label="Number of Video Frames", info="For >56 frames use local workstation!", value="24 - frames")
178
  with gr.Row():
179
- prompt_stage1 = gr.Textbox(label='Textual Prompt', placeholder="Ex: Dog running on the street.")
180
  with gr.Row():
181
  image_stage1 = gr.Image(label='Image Prompt (first select Image-to-Video model from advanced options to enable image upload)', show_label=True, scale=1, show_download_button=False, interactive=False)
182
  with gr.Column():
 
1
  import torch
2
+ # torch.jit.script = lambda f: f
3
  # General
4
  import os
5
  from os.path import join as opj
6
  import argparse
7
  import datetime
8
  from pathlib import Path
9
+ # import spaces
10
  import gradio as gr
11
  import tempfile
12
  import yaml
 
59
  # -------------------------
60
  # ----- Functionality -----
61
  # -------------------------
62
+ # @spaces.GPU(duration=120)
63
  def generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance, where_to_log=result_fol):
64
  now = datetime.datetime.now()
65
  name = prompt[:100].replace(" ", "_") + "_" + str(now.time()).replace(":", "_").replace(".", "_")
 
89
  video_path = opj(where_to_log, name+".mp4")
90
  return video_path
91
 
92
+ # @spaces.GPU(duration=400)
93
  def enhance(prompt, input_to_enhance, num_frames=None, image=None, model_name_stage1=None, model_name_stage2=None, seed=33, t=50, image_guidance=9.5, result_fol=result_fol):
94
  if input_to_enhance is None:
95
  input_to_enhance = generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance)
 
176
  with gr.Row():
177
  num_frames = gr.Dropdown(["24 - frames", "32 - frames", "40 - frames", "48 - frames", "56 - frames", "80 - recommended to run on local GPUs", "240 - recommended to run on local GPUs", "600 - recommended to run on local GPUs", "1200 - recommended to run on local GPUs", "10000 - recommended to run on local GPUs"], label="Number of Video Frames", info="For >56 frames use local workstation!", value="24 - frames")
178
  with gr.Row():
179
+ prompt_stage1 = gr.Textbox(label='Textual Prompt', placeholder="Ex: A man with yellow ballon head is riding a bike on the street of New York City", value="A man with yellow ballon head is riding a bike on the street of New York City")
180
  with gr.Row():
181
  image_stage1 = gr.Image(label='Image Prompt (first select Image-to-Video model from advanced options to enable image upload)', show_label=True, scale=1, show_download_button=False, interactive=False)
182
  with gr.Column():