fdfsdfs

app.py

@@ -13,12 +13,11 @@ question = "Qual é o maior planeta do sistema solar?"
 
 before = datetime.datetime.now()
 
-# Load model directly
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-6B-200K")
-model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-6B-200K")
+tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-1.5-6B-Chat")
+model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-1.5-6B-Chat")
 
 st.write('tokenizando...')
 prompt = "Hey, are you conscious? Can you talk to me?"