Update app.py

app.py

@@ -5,8 +5,8 @@ import torch.nn.functional as F
 import pandas as pd
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained('gpt2')
-model = AutoModelForCausalLM.from_pretrained('gpt2')
+tokenizer = AutoTokenizer.from_pretrained('mistralai/Mistral-7B-v0.1', device_map="auto", load_in_4bit=True)
+model = AutoModelForCausalLM.from_pretrained('mistralai/Mistral-7B-v0.1', device_map="auto", load_in_4bit=True)
 text1 = solara.reactive("Never gonna give you up, never gonna let you")
 @solara.component
 def Page():