Yardenfren commited on
Commit
856c348
·
verified ·
1 Parent(s): cb95c10

Update app_inference.py

Browse files
Files changed (1) hide show
  1. app_inference.py +4 -4
app_inference.py CHANGED
@@ -43,10 +43,6 @@ body {
43
  }
44
  """
45
 
46
- hf_token = os.getenv('HF_TOKEN')
47
- pipe = InferencePipeline(hf_token)
48
- app = InferenceUtil(hf_token)
49
-
50
  def get_choices(hf_token):
51
  api = HfApi(token=hf_token)
52
  choices = [
@@ -132,6 +128,10 @@ class InferenceUtil:
132
  raise type(e)(f'failed to update_model_info, due to: {e}')
133
 
134
 
 
 
 
 
135
 
136
  with gr.Blocks(css=css) as demo:
137
  with gr.Row(elem_classes="gr-row"):
 
43
  }
44
  """
45
 
 
 
 
 
46
  def get_choices(hf_token):
47
  api = HfApi(token=hf_token)
48
  choices = [
 
128
  raise type(e)(f'failed to update_model_info, due to: {e}')
129
 
130
 
131
+ hf_token = os.getenv('HF_TOKEN')
132
+ pipe = InferencePipeline(hf_token)
133
+ app = InferenceUtil(hf_token)
134
+
135
 
136
  with gr.Blocks(css=css) as demo:
137
  with gr.Row(elem_classes="gr-row"):