Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|