Hugging Face's logo

Spaces:
tonyassi
/
sketch-to-fashion-design
Running on Zero

App Files
tonyassi commited on
Commit
b8e5f1a
·
verified ·
1 Parent(s): c1a5c2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import gradio as gr
3
  from PIL import Image
4
  from diffusers import StableDiffusionImg2ImgPipeline
5
  import torch
 
6
 
7
  model_id_or_path = os.environ.get('MODEL')
8
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16,safety_checker=None, use_safetensors=True)
 
3
  from PIL import Image
4
  from diffusers import StableDiffusionImg2ImgPipeline
5
  import torch
6
+ import os
7
 
8
  model_id_or_path = os.environ.get('MODEL')
9
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16,safety_checker=None, use_safetensors=True)