jiwan-chung commited on
Commit
a2217bb
·
1 Parent(s): 5a61cb9

fix window size

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. run.py +3 -3
app.py CHANGED
@@ -37,7 +37,6 @@ examples = [[
37
  v,
38
  'My favourite recipe:',
39
  20,
40
- 10,
41
  False
42
  ] for v in images.values()]
43
 
 
37
  v,
38
  'My favourite recipe:',
39
  20,
 
40
  False
41
  ] for v in images.values()]
42
 
run.py CHANGED
@@ -125,10 +125,10 @@ class Runner:
125
  def __init__(self, inferer):
126
  self.inferer = inferer
127
 
128
- def __call__(self, inp, prompt, length, window_size, sample):
129
  # inp = inp.reshape((224, 224, 3))
130
  img = Image.fromarray(np.uint8(inp))
131
- text = self.inferer(img, prompt, length, window_size, sample=sample)
132
  return prompt, text
133
  # return inp, prompt, text
134
 
@@ -151,7 +151,7 @@ def launch(examples=None):
151
  inputs=[gr.components.Image(shape=(224, 224)),
152
  gr.components.Textbox(label='prompt'),
153
  gr.components.Slider(20, 120, step=1, label='length'),
154
- gr.components.Slider(10, 100, step=1, label='window_size'),
155
  gr.components.Checkbox(label='do sample')],
156
  outputs=[gr.components.Textbox(label='prompt'),
157
  gr.components.Textbox(label='generation')],
 
125
  def __init__(self, inferer):
126
  self.inferer = inferer
127
 
128
+ def __call__(self, inp, prompt, length, sample):
129
  # inp = inp.reshape((224, 224, 3))
130
  img = Image.fromarray(np.uint8(inp))
131
+ text = self.inferer(img, prompt, length, window_size=10, sample=sample)
132
  return prompt, text
133
  # return inp, prompt, text
134
 
 
151
  inputs=[gr.components.Image(shape=(224, 224)),
152
  gr.components.Textbox(label='prompt'),
153
  gr.components.Slider(20, 120, step=1, label='length'),
154
+ # gr.components.Slider(10, 100, step=1, label='window_size'),
155
  gr.components.Checkbox(label='do sample')],
156
  outputs=[gr.components.Textbox(label='prompt'),
157
  gr.components.Textbox(label='generation')],