changed to a new api endpoint

app.py

@@ -12,7 +12,6 @@ with open('style.css') as f:
 
 # Initialize the HuggingFace Inference Client
 text_client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.1")
-#image_client = Client("Boboiazumi/animagine-xl-3.1")
 image_client = Client("phenixrhyder/nsfw-waifu-gradio")
 
 def format_prompt_for_description(name, hair_color, personality, outfit_style, hobbies, favorite_food, background_story):
@@ -49,35 +48,16 @@ def generate_text(prompt, temperature=0.9, max_new_tokens=512, top_p=0.95, repet
         st.error(f"Error generating text: {e}")
         return ""
 
+# Updated part for the new API
 def generate_image(prompt):
     try:
         result = image_client.predict(
-            prompt=prompt,
-            negative_prompt="",
-            seed=0,
-            custom_width=1024,
-            custom_height=1024,
-            guidance_scale=7.0,
-            num_inference_steps=28,
-            sampler="Euler a",
-            aspect_ratio_selector="896 x 1152",
-            style_selector="(None)",
-            quality_selector="Standard v3.1",
-            use_upscaler=False,
-            upscaler_strength=0.55,
-            upscale_by=1.5,
-            add_quality_tags=True,
-            isImg2Img=False,
-            img_path=None,
-            img2img_strength=0.65,
+            param_0=prompt,
             api_name="/predict"
         )
         # Process and display the result
         if result:
-            images = []
-            for image_data in result[0]:
-                images.append(image_data['image'])
-            return images
+            return [result]  # Assuming the API returns a single image path as a result
         else:
             st.error("Unexpected result format from the Gradio API.")
             return None