helenai commited on
Commit
47b430f
1 Parent(s): fb77874

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -18,6 +18,7 @@ sys.path.append(str(test_path))
18
  from test_modeling import *
19
  from test_stable_diffusion import *
20
 
 
21
 
22
  def generate_model_list():
23
  tests = []
@@ -41,7 +42,7 @@ def generate_model_list():
41
  print(cls.SUPPORTED_ARCHITECTURES_WITH_ATTENTION)
42
  d[task] = cls.SUPPORTED_ARCHITECTURES_WITH_ATTENTION + cls.SUPPORTED_ARCHITECTURES_WITH_HIDDEN_STATES
43
 
44
- with open("supported_models.md", "w") as f:
45
  f.write(f"Updated at {datetime.now().strftime('%d %B %Y')}\n\n")
46
 
47
  summary = []
@@ -62,14 +63,14 @@ def generate_model_list():
62
  f.write(f"Total unique architectures: {num_total_archs}\n\n")
63
  f.write(f"Total validated architecture/task combinations: {len(all_archs)}\n\n")
64
 
65
- return Path("supported_models.md").read_text()
66
 
67
 
68
  demo = gr.Interface(
69
  fn=generate_model_list,
70
  title="List of validated architectures for optimum[openvino]",
71
  inputs=[],
72
- outputs=[gr.Markdown()],
73
  allow_flagging="never",
74
  )
75
  demo.launch(server_name="0.0.0.0")
 
18
  from test_modeling import *
19
  from test_stable_diffusion import *
20
 
21
+ RESULT_FILE = "supported_models.md"
22
 
23
  def generate_model_list():
24
  tests = []
 
42
  print(cls.SUPPORTED_ARCHITECTURES_WITH_ATTENTION)
43
  d[task] = cls.SUPPORTED_ARCHITECTURES_WITH_ATTENTION + cls.SUPPORTED_ARCHITECTURES_WITH_HIDDEN_STATES
44
 
45
+ with open(RESULT_FILE, "w") as f:
46
  f.write(f"Updated at {datetime.now().strftime('%d %B %Y')}\n\n")
47
 
48
  summary = []
 
63
  f.write(f"Total unique architectures: {num_total_archs}\n\n")
64
  f.write(f"Total validated architecture/task combinations: {len(all_archs)}\n\n")
65
 
66
+ return Path(RESULT_FILE).read_text(), RESULT_FILE
67
 
68
 
69
  demo = gr.Interface(
70
  fn=generate_model_list,
71
  title="List of validated architectures for optimum[openvino]",
72
  inputs=[],
73
+ outputs=[gr.Markdown(), gr.File()],
74
  allow_flagging="never",
75
  )
76
  demo.launch(server_name="0.0.0.0")