Update app.py

app.py

@@ -161,7 +161,7 @@ except Exception as e:
 @spaces.GPU
 def generate_image(prompt, input_image, lora_weight, guidance, downsampling_factor, weight, seed, width, height, batch_size, steps, progress=gr.Progress(track_tqdm=True)):
    try:
-       with torch.inference_mode():
+       with torch.no_grad():
            logger.info(f"Iniciando geração com prompt: {prompt}")
            
            # Codificar texto