Update app.py

app.py

@@ -29,7 +29,7 @@ class IntelChatBot:
         prompt = f"### System:\n {self.system_message}\n ### User:\n{user_message}\n### System:\n"
         return prompt
 
-    def predict(self, user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample):
+    def Neuro(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)
@@ -50,14 +50,14 @@ class IntelChatBot:
         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_neuro(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample):
     Intel_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 = Intel_bot.predict(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
+    response = Intel_bot.neuro(user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample)
     return response
 
 Intel_bot = IntelChatBot(model, tokenizer)
@@ -83,7 +83,7 @@ with gr.Blocks(theme = "ParityError/Anime") as demo:
     output_text = gr.Textbox(label="🧠🤌🏻NeuralChat 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_neuro(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty, do_sample)
 
     submit_button.click(
         process,