test

app.py

@@ -50,7 +50,7 @@ def generate_image_with_refinement(prompt):
     high_noise_frac = 0.8
 
     # run both experts
-    image = pipe(prompt=prompt).images[0]
+    image = base(prompt=prompt).images[0]
     # Save the image as before
     unique_id = str(uuid.uuid4())
     image_path = f"generated_images_refined/{unique_id}.jpeg"