Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -324,12 +324,7 @@ with gr.Blocks(theme="YTheme/Gradient", css=css) as demo:
|
|
324 |
)
|
325 |
|
326 |
if __name__ == "__main__":
|
327 |
-
|
328 |
-
# The `if torch.cuda.is_available():` block handles model loading for GPU.
|
329 |
-
# A CPU fallback for inference would require a CPU-compatible model or different handling in `generate`.
|
330 |
-
# The provided code primarily targets GPU.
|
331 |
-
# Added a basic placeholder image generation in `generate` if pipes are not loaded.
|
332 |
-
# Also need `ImageDraw` for that.
|
333 |
from PIL import ImageDraw # Add ImageDraw import for CPU placeholder
|
334 |
|
335 |
demo.queue(max_size=20).launch(ssr_mode=True, show_error=True, share=True)
|
|
|
324 |
)
|
325 |
|
326 |
if __name__ == "__main__":
|
327 |
+
|
|
|
|
|
|
|
|
|
|
|
328 |
from PIL import ImageDraw # Add ImageDraw import for CPU placeholder
|
329 |
|
330 |
demo.queue(max_size=20).launch(ssr_mode=True, show_error=True, share=True)
|