zenafey commited on
Commit
3833472
1 Parent(s): 64000d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -50,6 +50,10 @@ class Prodia:
50
  response = self._get(f"{self.base}/sd/models")
51
  return response.json()
52
 
 
 
 
 
53
  def _post(self, url, params):
54
  headers = {
55
  **self.headers,
@@ -325,6 +329,9 @@ with gr.Blocks(css=css) as demo:
325
  cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
326
  seed = gr.Number(label="Seed", value=-1)
327
 
 
 
 
328
  with gr.Column(scale=2):
329
  image_output = gr.Gallery(value=["https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png"], preview=True)
330
 
 
50
  response = self._get(f"{self.base}/sd/models")
51
  return response.json()
52
 
53
+ def list_loras(self):
54
+ response = self._get(f"{self.base}/sd/loras")
55
+ return response.json()
56
+
57
  def _post(self, url, params):
58
  headers = {
59
  **self.headers,
 
329
  cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
330
  seed = gr.Number(label="Seed", value=-1)
331
 
332
+ with gr.Tab("Lora"):
333
+ first_btn = gr.Button(prodia_client.list_loras()[0])
334
+
335
  with gr.Column(scale=2):
336
  image_output = gr.Gallery(value=["https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png"], preview=True)
337