Update app.py

app.py

@@ -22,15 +22,15 @@ def infer(prompt, init_image):
     init_image = Image.open(init_image).convert("RGB")
     init_image = init_image.resize((128, 128))
 
-    with torch.autocast('cuda'):
-        res = pipe.train(
-            prompt,
-            init_image,
-            guidance_scale=7.5,
-            num_inference_steps=50,
-            generator=generator,
-            text_embedding_optimization_steps=100,
-            model_fine_tuning_optimization_steps=500)
+   
+    res = pipe.train(
+        prompt,
+        init_image,
+        guidance_scale=7.5,
+        num_inference_steps=50,
+        generator=generator,
+        text_embedding_optimization_steps=100,
+        model_fine_tuning_optimization_steps=500)
     
     #with torch.no_grad():
     #    torch.cuda.empty_cache()