Update app.py

app.py

@@ -1589,6 +1589,9 @@ def maybe_upload_batch_set(pred_json_path):
             print(f'Failed to save to repo: {DATA_SET_REPO_PATH}|{str(e)}')
 
 
+def free_form_prompt(prompt, history=None, system_prompt=None):
+    return prompt
+
 def batch_inference(
         files: Union[str, List[str]], 
         prompt_mode: str,
@@ -1633,9 +1636,10 @@ def batch_inference(
         prompt_format_fn = chatml_format
     elif prompt_mode == 'few-shot':
         from functools import partial
-        prompt_format_fn = partial(
-            chatml_format, include_end_instruct=False
-        )
+        # prompt_format_fn = partial(
+        #     chatml_format, include_end_instruct=False
+        # )
+        prompt_format_fn = free_form_prompt 
     else:
         raise gr.Error(f'Wrong mode {prompt_mode}')
 
@@ -1923,10 +1927,10 @@ def launch():
         demo.launch(server_port=PORT, show_api=False, allowed_paths=["seal_logo.png"])
 
 
-def main():
+# def main():
 
-    launch()
+#     launch()
 
 
 if __name__ == "__main__":
-    main()
+    launch()