Spaces:
Running
on
Zero
Running
on
Zero
Update app_inference.py
Browse files- 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"):
|