Update app.py

app.py

@@ -1,17 +1,17 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 import streamlit as st
 from huggingface_hub import login
 import pandas as pd
 from threading import Thread
-import accelerate
 
-# Token Secret of Hugging Face
+# Token Secret de Hugging Face
 huggingface_token = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
 login(huggingface_token)
 
+# Cambiar a la versión Meta Llama 3.1 3B
+model_id = "meta-llama/Meta-Llama-3.1-3B"
 
-# Cargar el modelo y el tokenizer
-model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+# Cargar el tokenizador y el modelo
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
 tokenizer.pad_token = tokenizer.eos_token
@@ -44,7 +44,7 @@ def generate_response(input_text, temperature=0.7, max_new_tokens=100):
         yield "".join(outputs)
 
 def main():
-    st.title("Chat con Meta Llama 3.1 8B")
+    st.title("Chat con Meta Llama 3.1 3B")
     
     # Paso 1: Subir el archivo CSV
     uploaded_file = st.file_uploader("Por favor, sube un archivo CSV para iniciar:", type=["csv"])