Spaces:
Sleeping
Sleeping
Commit ·
8ebafab
1
Parent(s): cc4a7cb
Final
Browse files
app.py
CHANGED
|
@@ -153,6 +153,7 @@ textarea:focus, input[type="text"]:focus {
|
|
| 153 |
"""
|
| 154 |
|
| 155 |
|
|
|
|
| 156 |
def predict(prompt, opt_a, opt_b, opt_c, opt_d, opt_e):
|
| 157 |
"""Tokenize, run model forward pass, and return ranked answers."""
|
| 158 |
if not prompt.strip():
|
|
|
|
| 153 |
"""
|
| 154 |
|
| 155 |
|
| 156 |
+
@spaces.GPU
|
| 157 |
def predict(prompt, opt_a, opt_b, opt_c, opt_d, opt_e):
|
| 158 |
"""Tokenize, run model forward pass, and return ranked answers."""
|
| 159 |
if not prompt.strip():
|