Update
Browse files
app.py
CHANGED
@@ -65,8 +65,8 @@ def compute(image, prompt, model_name, sampler):
|
|
65 |
else:
|
66 |
if not model_name:
|
67 |
raise gr.Error('Models not loaded yet')
|
68 |
-
|
69 |
-
output = 'output'
|
70 |
logging.info('output="%s"', output)
|
71 |
|
72 |
width, height = image.size
|
|
|
65 |
else:
|
66 |
if not model_name:
|
67 |
raise gr.Error('Models not loaded yet')
|
68 |
+
output = models.generate(model_name, sampler, image, prompt)
|
69 |
+
# output = 'output'
|
70 |
logging.info('output="%s"', output)
|
71 |
|
72 |
width, height = image.size
|