Update app.py

app.py

@@ -38,25 +38,40 @@ pipe.enable_attention_slicing()
 # ) 
 
 pipe.unet.load_attn_procs(
-    ".",
-    weight_name="pytorch_custom_diffusion_weights.bin"
+"./pytorch_custom_diffusion_weights.bin"
 )
 
 print("Pipeline loaded")
 
+# def generate(prompt, steps, guidance):
+
+#     print("Generating...")
+    
+#     image = pipe(
+#         prompt,
+#         num_inference_steps=steps,
+#         guidance_scale=guidance,
+#         eta=1
+#     ).images[0]
+
+#     print("Done")
+    
+#     return image
+
 def generate(prompt, steps, guidance):
 
     print("Generating...")
-    
-    image = pipe(
+
+    result = pipe(
         prompt,
-        num_inference_steps=steps,
-        guidance_scale=guidance,
-        eta=1
-    ).images[0]
+        num_inference_steps=int(steps),
+        guidance_scale=float(guidance)
+    )
+
+    print("RESULT:", result)
+
+    image = result.images[0]
 
-    print("Done")
-    
     return image