Update app.py

app.py

@@ -9,7 +9,7 @@ api_key = os.getenv('HF_API_KEY')
 from gradio_client import Client
 
 
-def predict():
+def predict(prompt):
     client = Client("fffiloni/sd-xl-custom-model")
     result = client.predict(
     		custom_model="lichorosario/dott_remastered_style_lora_sdxl",
@@ -20,7 +20,7 @@ def predict():
     result = client.predict(
 		custom_model="lichorosario/dott_remastered_style_lora_sdxl",
 		weight_name="dott_style.safetensors",
-		prompt="house interior. two vanishing points, one at the left and one at the right.  Cartoon background. Curved perspective. Retro cartoon scene. twisted objects. simple irregular shapes. Cartoon Expressionism.foreshortening. wide angle.  lens distortion. converging lines",
+		prompt=prompt,
 		inf_steps=25,
 		guidance_scale=12,
 		width=1024,
@@ -32,12 +32,11 @@ def predict():
 
     return result
 
-
-demo = gr.Interface(
-    fn=predict,
-    inputs=[],
-    outputs=["image", "text"],
-)
+with gr.Blocks() as demo:
+    inputs = gr.Textbox(label="Prompt")
+    outputs = [gr.Image(label="Image"), gr.Textbox(label="Seed")]
+    greet_btn = gr.Button("Generate")
+    greet_btn.click(fn=predict, inputs=inputs, outputs=outputs, api_name="predict")
 
 demo.launch()