Update app.py
Browse files
app.py
CHANGED
|
@@ -38,11 +38,6 @@ def train(prompt, init_image, trn_text, trn_steps):
|
|
| 38 |
with torch.no_grad():
|
| 39 |
torch.cuda.empty_cache()
|
| 40 |
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
return "Training is finished !"
|
| 47 |
|
| 48 |
def generate(prompt, init_image, trn_text, trn_steps):
|
|
|
|
| 38 |
with torch.no_grad():
|
| 39 |
torch.cuda.empty_cache()
|
| 40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
return "Training is finished !"
|
| 42 |
|
| 43 |
def generate(prompt, init_image, trn_text, trn_steps):
|