amildravid4292 commited on
Commit
3170f91
·
verified ·
1 Parent(s): 97cf55a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -213,13 +213,13 @@ class main():
213
 
214
  @spaces.GPU
215
  def sample_then_run(self):
216
- sample_model()
217
  prompt = "sks person"
218
  negative_prompt = "low quality, blurry, unfinished, nudity, weapon"
219
  seed = 5
220
  cfg = 3.0
221
  steps = 25
222
- image = inference( prompt, negative_prompt, cfg, steps, seed)
223
  torch.save(self.network.proj, "model.pt" )
224
  return image, "model.pt"
225
 
 
213
 
214
  @spaces.GPU
215
  def sample_then_run(self):
216
+ self.sample_model()
217
  prompt = "sks person"
218
  negative_prompt = "low quality, blurry, unfinished, nudity, weapon"
219
  seed = 5
220
  cfg = 3.0
221
  steps = 25
222
+ image = self.inference( prompt, negative_prompt, cfg, steps, seed)
223
  torch.save(self.network.proj, "model.pt" )
224
  return image, "model.pt"
225