tianleliphoebe commited on
Commit
3bb2685
1 Parent(s): 340b8c3
Files changed (2) hide show
  1. app.py +1 -1
  2. model/model_manager.py +1 -1
app.py CHANGED
@@ -100,4 +100,4 @@ if __name__ == "__main__":
100
 
101
  elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir)
102
  demo = build_combine_demo(models, elo_results_file, leaderboard_table_file)
103
- demo.queue(max_size=20).launch(server_port=server_port, root_path=ROOT_PATH)
 
100
 
101
  elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir)
102
  demo = build_combine_demo(models, elo_results_file, leaderboard_table_file)
103
+ demo.queue(max_size=20).launch(server_port=server_port, root_path=ROOT_PAT)
model/model_manager.py CHANGED
@@ -53,7 +53,7 @@ class ModelManager:
53
  results = [future.result() for future in futures]
54
  return results[0], results[1]
55
 
56
- @spaces.GPU(duration=150)
57
  def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
58
  pipe = self.load_model_pipe(model_name)
59
  result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)
 
53
  results = [future.result() for future in futures]
54
  return results[0], results[1]
55
 
56
+ @spaces.GPU(duration=200)
57
  def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
58
  pipe = self.load_model_pipe(model_name)
59
  result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)