Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import time
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
-prompt=''' Review the Following text as a human, who is tasked to Extract `Code Snippets` :
+prompt=''' Review the Following text as a human, who is tasked to Extract `Code Snippets`.This is very important for my career:
 
 "/{}/"
 '''
@@ -11,8 +11,12 @@ replacements = {
     '<': 'LessThanSign'
 }
 
-model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
-tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
+model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+#model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
+#tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
 
 def chatbot_model(m,history):
     for char, replacement in replacements.items():