kaveh commited on
Commit
7fab264
1 Parent(s): 3ae3bd3

removed log reading textbox

Browse files
Files changed (1) hide show
  1. app.py +1 -28
app.py CHANGED
@@ -4,32 +4,7 @@ import sys
4
 
5
  generator = DiffusionPipeline.from_pretrained("kaveh/wsi_generator")
6
 
7
- class Logger:
8
- def __init__(self, filename):
9
- self.terminal = sys.stdout
10
- self.log = open(filename, "w")
11
-
12
- def write(self, message):
13
- self.terminal.write(message)
14
- self.log.write(message)
15
-
16
- def flush(self):
17
- self.terminal.flush()
18
- self.log.flush()
19
-
20
- def isatty(self):
21
- return False
22
-
23
- sys.stdout = Logger("output.log")
24
-
25
- def read_logs():
26
- sys.stdout.flush()
27
- with open("output.log", "r") as f:
28
- return f.read()
29
-
30
-
31
-
32
- def generate(n_samples=1, progress=gr.Progress()):
33
  images = []
34
  for i in range(n_samples):
35
  image = generator().images[0]
@@ -46,7 +21,5 @@ with gr.Blocks() as demo:
46
  label="Generated images", show_label=False, elem_id="gallery").style(columns=[2], rows=[2], object_fit="contain", height="auto", preview=True)
47
 
48
  btn.click(generate, n_s, gallery)
49
- logs = gr.Textbox().style(container=False)
50
- demo.load(read_logs, None, logs, every=1)
51
 
52
  demo.launch()
 
4
 
5
  generator = DiffusionPipeline.from_pretrained("kaveh/wsi_generator")
6
 
7
+ def generate(n_samples=1):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  images = []
9
  for i in range(n_samples):
10
  image = generator().images[0]
 
21
  label="Generated images", show_label=False, elem_id="gallery").style(columns=[2], rows=[2], object_fit="contain", height="auto", preview=True)
22
 
23
  btn.click(generate, n_s, gallery)
 
 
24
 
25
  demo.launch()