kingpreyansh commited on
Commit
d92699e
1 Parent(s): f5724f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -1,13 +1,14 @@
1
  import os
2
-
3
  import json
4
  import requests
5
  import io
6
  import base64
 
7
  import gradio as gr
8
-
9
  from PIL import Image, PngImagePlugin
10
 
 
11
  model = "/home/user/app/sd/stable-diffusion-webui/models/Stable-diffusion/"
12
 
13
  #os.system(python /home/user/app/sd/stable-diffusion-webui/webui.py)
@@ -19,6 +20,8 @@ payload = {
19
  "steps": 5
20
  }
21
 
 
 
22
  response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
23
 
24
  r = response.json()
@@ -35,15 +38,14 @@ for i in r['images']:
35
  pnginfo.add_text("parameters", response2.json().get("info"))
36
  image.save('output.png', pnginfo=pnginfo)
37
 
38
- import psutil
39
-
40
 
41
- def current_working_directoy(cwd, ldir):
42
- return os.getcwd(), psutil.net_connections()
43
 
44
  iface = gr.Interface(fn=current_working_directoy,
45
 
46
- inputs=["text", "text"], outputs=["text", "text"])
47
 
48
 
49
  iface.launch()
 
1
  import os
2
+ import psutil
3
  import json
4
  import requests
5
  import io
6
  import base64
7
+ from time import sleep
8
  import gradio as gr
 
9
  from PIL import Image, PngImagePlugin
10
 
11
+
12
  model = "/home/user/app/sd/stable-diffusion-webui/models/Stable-diffusion/"
13
 
14
  #os.system(python /home/user/app/sd/stable-diffusion-webui/webui.py)
 
20
  "steps": 5
21
  }
22
 
23
+
24
+ sleep(5)
25
  response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
26
 
27
  r = response.json()
 
38
  pnginfo.add_text("parameters", response2.json().get("info"))
39
  image.save('output.png', pnginfo=pnginfo)
40
 
41
+ ##############################################################################################
 
42
 
43
+ def current_working_directoy(cwd, ldir, python_path):
44
+ return os.getcwd(), psutil.net_connections(), os.listdir('/usr/bin')
45
 
46
  iface = gr.Interface(fn=current_working_directoy,
47
 
48
+ inputs=["text", "text", "text"], outputs=["text", "text", "text"])
49
 
50
 
51
  iface.launch()