muellerzr HF staff commited on
Commit
562c3cb
1 Parent(s): fc22c78

Check for if discussion is disabled

Browse files
Files changed (1) hide show
  1. src/app.py +5 -1
src/app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import pandas as pd
3
  from hub_utils import check_for_discussion, report_results
4
  from model_utils import calculate_memory, get_model
 
5
 
6
 
7
  # We need to store them as globals because gradio doesn't have a way for us to pass them in to the button
@@ -11,7 +12,10 @@ MODEL = None
11
  def get_results(model_name: str, library: str, options: list, access_token: str):
12
  global MODEL
13
  MODEL = get_model(model_name, library, access_token)
14
- has_discussion = check_for_discussion(model_name)
 
 
 
15
  title = f"## Memory usage for '{model_name}'"
16
  data = calculate_memory(MODEL, options)
17
  return [title, gr.update(visible=True, value=pd.DataFrame(data)), gr.update(visible=not has_discussion)]
 
2
  import pandas as pd
3
  from hub_utils import check_for_discussion, report_results
4
  from model_utils import calculate_memory, get_model
5
+ from huggingface_hub.utils import HfHubHTTPError
6
 
7
 
8
  # We need to store them as globals because gradio doesn't have a way for us to pass them in to the button
 
12
  def get_results(model_name: str, library: str, options: list, access_token: str):
13
  global MODEL
14
  MODEL = get_model(model_name, library, access_token)
15
+ try:
16
+ has_discussion = check_for_discussion(model_name)
17
+ except HfHubHTTPError:
18
+ has_discussion = True
19
  title = f"## Memory usage for '{model_name}'"
20
  data = calculate_memory(MODEL, options)
21
  return [title, gr.update(visible=True, value=pd.DataFrame(data)), gr.update(visible=not has_discussion)]