Eyalgut commited on
Commit
d1bb2c2
1 Parent(s): 7465a1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@ import torch
3
  import numpy as np
4
  import diffusers
5
  import os
 
6
  hf_token = os.environ.get("HF_TOKEN")
7
  from diffusers import StableDiffusionXLInpaintPipeline, DDIMScheduler, UNet2DConditionModel
8
 
@@ -68,7 +69,7 @@ def read_content(file_path: str) -> str:
68
 
69
  return content
70
 
71
- def predict(dict, prompt="", negative_prompt="", guidance_scale=5, steps=30, strength=1.0, scheduler="EulerDiscreteScheduler"):
72
  if negative_prompt == "":
73
  negative_prompt = None
74
 
 
3
  import numpy as np
4
  import diffusers
5
  import os
6
+ from PIL import Image
7
  hf_token = os.environ.get("HF_TOKEN")
8
  from diffusers import StableDiffusionXLInpaintPipeline, DDIMScheduler, UNet2DConditionModel
9
 
 
69
 
70
  return content
71
 
72
+ def predict(dict, prompt="", negative_prompt="", guidance_scale=5, steps=30, strength=1.0):
73
  if negative_prompt == "":
74
  negative_prompt = None
75