Upload folder using huggingface_hub

app.py

@@ -38,12 +38,18 @@ def fn_to_str(x):
 
 
 def get_gpt_guess(clues, ask):
-    system_message = """You are an expert mathematician. You will be given a list of input and outputs to a function and your job is to guess the output of the function on an unknown input.
+    system_message = """You are an expert mathematician. You will be given a list of input and outputs to a function and your job is to guess the output of the function on an unknown input. If you cannot determine the answer, respond with 0.
     For example:
     f(2) = 6
     f(7) = 11
     What is f(3)?
-    Answer: 7"""
+    Answer: 7
+    
+    For example:
+    f(1) = 17
+    What is f(6)?
+    Answer: 0"""
+
 
     user_message = "\n".join(map(fn_to_str, clues)) + f"\nWhat is f({ask})?\nAnswer:"
     return chat([{"role": "system", "content": system_message}, {"role": "user", "content": user_message}], model=model)
@@ -79,8 +85,12 @@ with gr.Blocks() as demo:
     def submit_guess(value, history, clue_values):
         history.append(value)
 
-        gpt_guess_value = get_gpt_guess(clue_values, ask)
-        result = check_winner(int(gpt_guess_value), int(value), challenge(ask))
+        try:
+            gpt_guess_value = int(get_gpt_guess(clue_values, ask))
+        except Exception:
+            gpt_guess_value = 0
+
+        result = check_winner(gpt_guess_value, int(value), challenge(ask))
 
         clue = random.choice(list(available_numbers))
         available_numbers.remove(clue)
@@ -90,7 +100,7 @@ with gr.Blocks() as demo:
         return {
             guesses: history,
             history_box: ", ".join(history),
-            gpt_guess: gpt_guess_value,
+            gpt_guess: str(gpt_guess_value),
             clues: clue_values,
             title_text: title_text.value + result,
             clues_box: f"# Clues\n```\n{clue_str}\n```",