Spaces:
Running
Running
Update
Browse files
app.py
CHANGED
@@ -98,7 +98,7 @@ def compute(image, prompt, model_name, sampler):
|
|
98 |
|
99 |
def warmup(model_name):
|
100 |
image = PIL.Image.new('RGB', [1, 1])
|
101 |
-
_ = compute(image, '', model_name, 'greedy')
|
102 |
|
103 |
|
104 |
def reset():
|
@@ -161,7 +161,7 @@ def create_app():
|
|
161 |
gr.on(
|
162 |
[run.click, prompt.submit],
|
163 |
compute,
|
164 |
-
[image, prompt, model, sampler],
|
165 |
[highlighted_text, image, annotated_image],
|
166 |
)
|
167 |
clear.click(
|
|
|
98 |
|
99 |
def warmup(model_name):
|
100 |
image = PIL.Image.new('RGB', [1, 1])
|
101 |
+
_ = compute(image, '', model_name + "-text-model-q4_k_m.gguf", 'greedy')
|
102 |
|
103 |
|
104 |
def reset():
|
|
|
161 |
gr.on(
|
162 |
[run.click, prompt.submit],
|
163 |
compute,
|
164 |
+
[image, prompt, model + "-text-model-q4_k_m.gguf", sampler],
|
165 |
[highlighted_text, image, annotated_image],
|
166 |
)
|
167 |
clear.click(
|