Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -79,13 +79,11 @@ few_shot_selection = {
|
|
| 79 |
# Main Logic
|
| 80 |
# =====================
|
| 81 |
@spaces.GPU
|
| 82 |
-
def run_inference(model_id,
|
| 83 |
# Ensure data is ready and model is correct one
|
| 84 |
initialize_data()
|
| 85 |
load_model_if_needed(model_id)
|
| 86 |
|
| 87 |
-
if not split_name or not question_idx:
|
| 88 |
-
return "Select split and ID.", "", "", None, None, None
|
| 89 |
|
| 90 |
try:
|
| 91 |
question_idx = int(question_idx)
|
|
|
|
| 79 |
# Main Logic
|
| 80 |
# =====================
|
| 81 |
@spaces.GPU
|
| 82 |
+
def run_inference(model_id, question_idx, few_shots):
|
| 83 |
# Ensure data is ready and model is correct one
|
| 84 |
initialize_data()
|
| 85 |
load_model_if_needed(model_id)
|
| 86 |
|
|
|
|
|
|
|
| 87 |
|
| 88 |
try:
|
| 89 |
question_idx = int(question_idx)
|