Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import argparse
|
| 2 |
import json
|
| 3 |
import numpy as np
|
|
@@ -217,7 +218,7 @@ def generate_grid_images(gif_results):
|
|
| 217 |
plt.close(fig)
|
| 218 |
return image_array
|
| 219 |
|
| 220 |
-
|
| 221 |
def llm_call(question_prompt, model_name,
|
| 222 |
temperature=1, max_tokens=320,
|
| 223 |
top_p=1, n_samples=64, stop=None):
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import argparse
|
| 3 |
import json
|
| 4 |
import numpy as np
|
|
|
|
| 218 |
plt.close(fig)
|
| 219 |
return image_array
|
| 220 |
|
| 221 |
+
@spaces.GPU
|
| 222 |
def llm_call(question_prompt, model_name,
|
| 223 |
temperature=1, max_tokens=320,
|
| 224 |
top_p=1, n_samples=64, stop=None):
|