Update app.py

app.py

@@ -44,14 +44,14 @@ def remove_background_bria(input_path):
 
 # Fungsi untuk memproses gambar menggunakan prompt
 def text_to_image(prompt):
-    image = pipe(prompt).images[0]
+    image = sd_model(prompt).images[0]  # Use sd_model instead of pipe
     image_path = f"generated_images/{prompt.replace(' ', '_')}.png"
     image.save(image_path)
     return image, image_path
 
 def text_image_to_image(image, prompt):
     # Convert input image to PIL Image for processing
-    modified_image = pipe(prompt, init_image=image, strength=0.75).images[0]
+    modified_image = sd_model(prompt, init_image=image, strength=0.75).images[0]  # Use sd_model here too
     image_path = f"generated_images/{prompt.replace(' ', '_')}_modified.png"
     modified_image.save(image_path)
     return modified_image, image_path