Update space

app.py

@@ -8,6 +8,8 @@ class CustomPrompter(Prompter):
     def get_response(self, output: str) -> str:
         return output.split(self.template["response_split"])[1].strip().split("### Instruction:")[0]
 
+prompt_template_name = "alpaca"  # The prompt template to use, will default to alpaca.
+
 prompter = CustomPrompter(prompt_template_name)
 
 def tokenize(prompt, add_eos_token=True):