Update app.py

app.py

@@ -28,7 +28,7 @@ class TuluChatBot:
         prompt = f"<|assistant|>\n {self.system_message}\n\n <|user|>{user_message}\n\n<|assistant|>\n"
         return prompt
 
-    def predict(self, user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample):
+    def Tulu(self, user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample):
         prompt = self.format_prompt(user_message)
         inputs = self.tokenizer(prompt, return_tensors='pt', add_special_tokens=False)
         input_ids = inputs["input_ids"].to(self.model.device)
@@ -48,14 +48,14 @@ class TuluChatBot:
         response = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
         return response
 
-def gradio_predict(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample):
+def gradio_Tulu(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample):
     Tulu_bot.set_system_message(system_message)
     if not do_sample:
         max_length = 780
         temperature = 0.9  
         top_p = 0.9        
         repetition_penalty = 0.9  
-    response = Tulu_bot.predict(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
+    response = Tulu_bot.Tulu(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
     return response
 
 Tulu_bot = TuluChatBot(model, tokenizer)
@@ -81,7 +81,7 @@ with gr.Blocks(theme = "ParityError/Anime") as demo:
     output_text = gr.Textbox(label="🌷Tulu Response")
 
     def process(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample):
-        return gradio_predict(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample)
+        return gradio_Tulu(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample)
 
     submit_button.click(
         process,