Update app.py

app.py

@@ -113,8 +113,8 @@ STYLE = """
 from transformers import GPT2Tokenizer, AutoModelForCausalLM, AutoTokenizer
 import numpy as np
 
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+tokenizer = AutoTokenizer.from_pretrained("Locutusque/TinyMistral-248M-v2")
+model = AutoModelForCausalLM.from_pretrained("Locutusque/TinyMistral-248M-v2")
 tokenizer.pad_token_id = tokenizer.eos_token_id
 print("Loading finished.")
 def generate_html(token, node):