UAI-Software commited on
Commit
c388933
·
verified ·
1 Parent(s): 459f789

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -17,9 +17,10 @@ class EndpointHandler:
17
  # self.model= load_model(path)
18
  self.pipe = None
19
  self.modelName = ""
 
20
  baseReq = ImageRequest()
21
- baseReq.model = path
22
- print(f"Loading model: {path}")
23
  self.LoadModel(baseReq)
24
 
25
  def LoadModel(self, request):
 
17
  # self.model= load_model(path)
18
  self.pipe = None
19
  self.modelName = ""
20
+ base = "sd-community/sdxl-flash"
21
  baseReq = ImageRequest()
22
+ baseReq.model = base
23
+ print(f"Loading model: {base}")
24
  self.LoadModel(baseReq)
25
 
26
  def LoadModel(self, request):