Change Image Client

app.py

@@ -1,15 +1,5 @@
 from gradio_client import Client
 import gradio as gr
-import requests
-import io
-from PIL import Image
-
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-headers = {"Authorization": "Bearer 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"}
-
-def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.content
 
 
 def answer_question(question):
@@ -25,23 +15,28 @@ def answer_question(question):
 
 def fashion(question):
     output = answer_question(question)
-    
-    image_bytes = query({
-	"inputs": output,
-    })
-    print(image_bytes)
-    print(io.BytesIO(image_bytes))
-    image = Image.open(io.BytesIO(image_bytes))
-    image.save("img.txt")
-    return None, output, "img.txt"
+    image = client.predict(
+				output,	# str  in 'What you want the AI to generate. 77 Token Limit.' Textbox component
+				"blurry, cropped, nsfw, bad hands, disfigured",	# str  in 'What you Do Not want the AI to generate. 77 Token Limit' Textbox component
+				512,	# int | float (numeric value between 512 and 1024) in 'Height' Slider component
+				512,	# int | float (numeric value between 512 and 1024) in 'Width' Slider component
+				9,	# int | float (numeric value between 1 and 15) in 'scale' Slider component
+				25,	# int | float (numeric value between 25 and 100) in 'steps' Slider component
+				9326438724,	# int | float (numeric value between 1 and 9999999999999999) in 'seed' Slider component
+				"Yes",	# str  in 'Upscale?' Radio component
+				api_name="/predict"
+    )
+    print(image)
+    return None, output, image
 
 
 chat_client = Client("https://mosaicml-mpt-30b-chat.hf.space/", serialize = False)
 retrieval = Client("https://warlord-k-latestfashion.hf.space/")
+image_client = Client("https://manjushri-photoreal-v2.hf.space/")
 
 init_prompt ="## Instruction: You are a fashion expert and must return truthful responses as per the information. Do not answer with any information which isn't completely verified and correct. Do not lie. Do not present information where you don't know the answer. Do not include incorrect extra information. Your name is DesAIner. You are helpful and truthful. You provide fashion suggestions to shoppers based upon their latest purchases and the latest fahion trends."
 info="Latest Fashion Trends: \n"
-q_prompt="\n ##Instruction: Please provide fashion reccomendation for the following query in less than 3 lines: \n"  
+q_prompt="\n ##Instruction: Please provide fashion recommendation for the following query in less than 3 lines, specifying the clothing items and accesories: \n"  
 chatbot = [["", None]]
 
 with gr.Blocks() as demo: