abetlen commited on
Commit
4d699b1
·
1 Parent(s): 3d2ba8e
Files changed (1) hide show
  1. app.py +2 -2
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(