prithivMLmods commited on
Commit
9763b71
·
verified ·
1 Parent(s): b931dec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -324,12 +324,7 @@ with gr.Blocks(theme="YTheme/Gradient", css=css) as demo:
324
  )
325
 
326
  if __name__ == "__main__":
327
- # For CPU execution, model loading might take time or fail if not handled.
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)