Maximofn commited on
Commit
325c781
·
1 Parent(s): 11998d6

fix(src): :rocket: Remove hyvideo dependencies

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -8,8 +8,6 @@ import time
8
  from datetime import datetime
9
  import gradio as gr
10
 
11
- from hyvideo.config import parse_args
12
-
13
  @spaces.GPU
14
  def initialize_model(model):
15
  # quant_config = BitsAndBytesConfig(load_in_8bit=True)
@@ -71,7 +69,7 @@ def generate_video(
71
 
72
  return video_path
73
 
74
- def create_demo(model, save_path):
75
  pipeline = initialize_model(model)
76
 
77
  with gr.Blocks() as demo:
@@ -143,7 +141,6 @@ if __name__ == "__main__":
143
  os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
144
  server_name = os.getenv("SERVER_NAME", "0.0.0.0")
145
  server_port = int(os.getenv("SERVER_PORT", "8081"))
146
- args = parse_args()
147
  model = "hunyuanvideo-community/HunyuanVideo" # Actualizado el path del modelo
148
- demo = create_demo(model, args.save_path)
149
  demo.launch(server_name=server_name, server_port=server_port)
 
8
  from datetime import datetime
9
  import gradio as gr
10
 
 
 
11
  @spaces.GPU
12
  def initialize_model(model):
13
  # quant_config = BitsAndBytesConfig(load_in_8bit=True)
 
69
 
70
  return video_path
71
 
72
+ def create_demo(model):
73
  pipeline = initialize_model(model)
74
 
75
  with gr.Blocks() as demo:
 
141
  os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
142
  server_name = os.getenv("SERVER_NAME", "0.0.0.0")
143
  server_port = int(os.getenv("SERVER_PORT", "8081"))
 
144
  model = "hunyuanvideo-community/HunyuanVideo" # Actualizado el path del modelo
145
+ demo = create_demo(model)
146
  demo.launch(server_name=server_name, server_port=server_port)