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"):
|