Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ torch.hub.download_url_to_file(
|
|
26 |
"https://cdn.openai.com/dall-e-2/demos/text2im/astronaut/horse/photo/0.jpg", "astronaut.jpg"
|
27 |
)
|
28 |
|
29 |
-
|
30 |
def run(image: Union[str, PIL.Image.Image]) -> str:
|
31 |
if isinstance(image, str):
|
32 |
image = Image.open(image)
|
|
|
26 |
"https://cdn.openai.com/dall-e-2/demos/text2im/astronaut/horse/photo/0.jpg", "astronaut.jpg"
|
27 |
)
|
28 |
|
29 |
+
@spaces.GPU()
|
30 |
def run(image: Union[str, PIL.Image.Image]) -> str:
|
31 |
if isinstance(image, str):
|
32 |
image = Image.open(image)
|