Spaces:
Runtime error
Runtime error
Commit
·
a2217bb
1
Parent(s):
5a61cb9
fix window size
Browse files
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,
|
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')],
|