lichorosario commited on
Commit
c92760e
1 Parent(s): bfad2c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -8,10 +8,21 @@ api_key = os.getenv('HF_API_KEY')
8
 
9
  from gradio_client import Client
10
 
11
- client = Client("fffiloni/sd-xl-custom-model")
12
- result = client.predict(
13
- custom_model="lichorosario/dott_remastered_style_lora_sdxl",
14
- api_name="/load_model"
 
 
 
 
 
 
 
 
 
 
15
  )
16
- print(result)
 
17
 
 
8
 
9
  from gradio_client import Client
10
 
11
+
12
+ def predict():
13
+ client = Client("fffiloni/sd-xl-custom-model")
14
+ result = client.predict(
15
+ custom_model="lichorosario/dott_remastered_style_lora_sdxl",
16
+ api_name="/load_model"
17
+ )
18
+ return result
19
+
20
+
21
+ demo = gr.Interface(
22
+ fn=predict,
23
+ inputs=[],
24
+ outputs=["text"],
25
  )
26
+
27
+ demo.launch()
28