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(
|