Spaces:
Running
on
Zero
Running
on
Zero
tianleliphoebe
commited on
Commit
•
3bb2685
1
Parent(s):
340b8c3
modify
Browse files- app.py +1 -1
- 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=
|
|
|
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=
|
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)
|