Spaces:
Build error
Build error
Khalil
commited on
Commit
·
b969123
1
Parent(s):
35ed471
fix interface not lunching error 'debug=True'.
Browse files
app.py
CHANGED
|
@@ -9,8 +9,6 @@ if not Path('./Text2Punk-final-7.pt').exists() and not Path('./clip-final.pt').e
|
|
| 9 |
|
| 10 |
# plot
|
| 11 |
|
| 12 |
-
import matplotlib.pyplot as plt
|
| 13 |
-
import numpy as np
|
| 14 |
from PIL import Image
|
| 15 |
|
| 16 |
# gradio
|
|
@@ -35,9 +33,7 @@ def compose_predictions(images):
|
|
| 35 |
image_grid = Image.new("RGB", (b*w*4, h*4), color=0)
|
| 36 |
|
| 37 |
for i in range(b):
|
| 38 |
-
|
| 39 |
-
print(images[i].shape)
|
| 40 |
-
img_ = to_pil_image(images[i])
|
| 41 |
image_grid.paste(img_, (i*w*4, increased_h))
|
| 42 |
|
| 43 |
return image_grid
|
|
@@ -59,15 +55,16 @@ def run_inference(prompt, num_images=32, batch_size=32, num_preds=8):
|
|
| 59 |
|
| 60 |
|
| 61 |
outputs = [
|
| 62 |
-
gr.outputs.HTML(label=""), # To be used as title
|
| 63 |
-
gr.outputs.Image(label=
|
| 64 |
]
|
| 65 |
|
| 66 |
description = """
|
| 67 |
Text2Cryptopunks is an AI model that generates Cryptopunks images from text prompt:
|
| 68 |
"""
|
| 69 |
|
| 70 |
-
gr.Interface(
|
|
|
|
| 71 |
inputs=[gr.inputs.Textbox(label='type somthing like this : "An Ape CryptoPunk that has 2 Attributes, a Pigtails and a Medical Mask."')],
|
| 72 |
outputs=outputs,
|
| 73 |
title='Text2Cryptopunks',
|
|
@@ -76,7 +73,8 @@ gr.Interface(run_inference,
|
|
| 76 |
layout='vertical',
|
| 77 |
theme='huggingface',
|
| 78 |
examples=[['Cute Alien cryptopunk that has a 2 Attributes, a Pipe, and a Beanie.'], ['A low resolution photo of punky-looking Ape that has 2 Attributes, a Beanie, and a Medical Mask.']],
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
#
|
| 82 |
-
|
|
|
|
|
|
| 9 |
|
| 10 |
# plot
|
| 11 |
|
|
|
|
|
|
|
| 12 |
from PIL import Image
|
| 13 |
|
| 14 |
# gradio
|
|
|
|
| 33 |
image_grid = Image.new("RGB", (b*w*4, h*4), color=0)
|
| 34 |
|
| 35 |
for i in range(b):
|
| 36 |
+
img_ = to_pil_image(resize(images[i], 96))
|
|
|
|
|
|
|
| 37 |
image_grid.paste(img_, (i*w*4, increased_h))
|
| 38 |
|
| 39 |
return image_grid
|
|
|
|
| 55 |
|
| 56 |
|
| 57 |
outputs = [
|
| 58 |
+
gr.outputs.HTML(label="Output Title"), # To be used as title
|
| 59 |
+
gr.outputs.Image(type="pil", label="Output Image"),
|
| 60 |
]
|
| 61 |
|
| 62 |
description = """
|
| 63 |
Text2Cryptopunks is an AI model that generates Cryptopunks images from text prompt:
|
| 64 |
"""
|
| 65 |
|
| 66 |
+
gr.Interface(
|
| 67 |
+
fn=run_inference,
|
| 68 |
inputs=[gr.inputs.Textbox(label='type somthing like this : "An Ape CryptoPunk that has 2 Attributes, a Pigtails and a Medical Mask."')],
|
| 69 |
outputs=outputs,
|
| 70 |
title='Text2Cryptopunks',
|
|
|
|
| 73 |
layout='vertical',
|
| 74 |
theme='huggingface',
|
| 75 |
examples=[['Cute Alien cryptopunk that has a 2 Attributes, a Pipe, and a Beanie.'], ['A low resolution photo of punky-looking Ape that has 2 Attributes, a Beanie, and a Medical Mask.']],
|
| 76 |
+
enable_queue=True,
|
| 77 |
+
# allow_flagging=False,
|
| 78 |
+
# live=False,
|
| 79 |
+
# verbose=True,
|
| 80 |
+
).launch(debug=True)
|