pihull commited on
Commit
33de83d
·
verified ·
1 Parent(s): bb8417b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -79,13 +79,11 @@ few_shot_selection = {
79
  # Main Logic
80
  # =====================
81
  @spaces.GPU
82
- def run_inference(model_id, split_name, 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
- 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)