Upload folder using huggingface_hub
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def compute_loss(generated_features, content_features, style_features, alpha, be
|
|
| 58 |
style_loss += w_l * F.mse_loss(G, A)
|
| 59 |
return alpha * content_loss + beta * style_loss
|
| 60 |
|
| 61 |
-
@spaces.GPU(duration=
|
| 62 |
def inference(content_image, style_name, style_strength, output_quality, progress=gr.Progress(track_tqdm=True)):
|
| 63 |
yield None
|
| 64 |
print('-'*15)
|
|
|
|
| 58 |
style_loss += w_l * F.mse_loss(G, A)
|
| 59 |
return alpha * content_loss + beta * style_loss
|
| 60 |
|
| 61 |
+
@spaces.GPU(duration=6)
|
| 62 |
def inference(content_image, style_name, style_strength, output_quality, progress=gr.Progress(track_tqdm=True)):
|
| 63 |
yield None
|
| 64 |
print('-'*15)
|