Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -127,6 +127,7 @@ enable_btn = gr.Button(interactive=True)
|
|
| 127 |
disable_btn = gr.Button(interactive=False)
|
| 128 |
|
| 129 |
# Model loading function
|
|
|
|
| 130 |
def load_model(model_path):
|
| 131 |
global tokenizer, model, processor, context_len
|
| 132 |
|
|
|
|
| 127 |
disable_btn = gr.Button(interactive=False)
|
| 128 |
|
| 129 |
# Model loading function
|
| 130 |
+
@spaces.GPU
|
| 131 |
def load_model(model_path):
|
| 132 |
global tokenizer, model, processor, context_len
|
| 133 |
|