Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,11 +16,12 @@ model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=huggingf
|
|
| 16 |
|
| 17 |
|
| 18 |
|
| 19 |
-
|
|
|
|
| 20 |
|
| 21 |
-
if
|
| 22 |
# Cargar el CSV y extraer la columna 'job_titles'
|
| 23 |
-
df = pd.read_csv(
|
| 24 |
job_titles = df['job_titles'].tolist()
|
| 25 |
|
| 26 |
# Crear la lista de job titles en formato de texto para el prompt
|
|
@@ -54,6 +55,4 @@ if csv_file:
|
|
| 54 |
st.write(response)
|
| 55 |
|
| 56 |
# Limpiar memoria después de la inferencia
|
| 57 |
-
torch.cuda.empty_cache() # Si estás usando una GPU
|
| 58 |
-
|
| 59 |
-
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
|
| 19 |
+
# Usar Streamlit para cargar el archivo CSV
|
| 20 |
+
uploaded_file = st.file_uploader("Por favor sube un archivo CSV:", type="csv")
|
| 21 |
|
| 22 |
+
if uploaded_file is not None:
|
| 23 |
# Cargar el CSV y extraer la columna 'job_titles'
|
| 24 |
+
df = pd.read_csv(uploaded_file)
|
| 25 |
job_titles = df['job_titles'].tolist()
|
| 26 |
|
| 27 |
# Crear la lista de job titles en formato de texto para el prompt
|
|
|
|
| 55 |
st.write(response)
|
| 56 |
|
| 57 |
# Limpiar memoria después de la inferencia
|
| 58 |
+
torch.cuda.empty_cache() # Si estás usando una GPU
|
|
|
|
|
|