Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -182,6 +182,7 @@ def get_conv_log_filename():
|
|
| 182 |
return name
|
| 183 |
|
| 184 |
# Inference function
|
|
|
|
| 185 |
def run_inference(prompt, image, temperature=0.2, top_p=0.95, max_tokens=512):
|
| 186 |
global model, tokenizer, processor
|
| 187 |
|
|
|
|
| 182 |
return name
|
| 183 |
|
| 184 |
# Inference function
|
| 185 |
+
@spaces.GPU
|
| 186 |
def run_inference(prompt, image, temperature=0.2, top_p=0.95, max_tokens=512):
|
| 187 |
global model, tokenizer, processor
|
| 188 |
|