JaphetHernandez commited on
Commit
0651e4c
·
verified ·
1 Parent(s): 2096cbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -16,11 +16,12 @@ model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=huggingf
16
 
17
 
18
 
19
- csv_file = st.text_input("Por favor introduce la ruta del archivo CSV:")
 
20
 
21
- if csv_file:
22
  # Cargar el CSV y extraer la columna 'job_titles'
23
- df = pd.read_csv(csv_file)
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