Eduarr commited on
Commit
b5947af
·
1 Parent(s): 265c2b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -13
app.py CHANGED
@@ -1,5 +1,20 @@
1
  import gradio as gr
2
  import openai
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
 
4
  def generate_completion(user_prompt):
5
  hidden_context = " "
@@ -7,22 +22,22 @@ def generate_completion(user_prompt):
7
  response = openai.Completion.create(
8
  model="davinci:ft-topwow-llc:rrhh-2023-09-26-22-30-27",
9
  prompt=prompt,
10
- max_tokens=30,
11
  temperature=0,
12
  stop=["_END"]
13
  )
14
 
15
-
16
  return response.choices[0].text.strip()
17
- iface = gr.Interface(fn=generate_completion,
18
- inputs=gr.inputs.Textbox(lines=5, placeholder=''),
19
- outputs='text',
20
- title="Predictor de renuncia",
21
- description="Plataforma para profesionales de los recursos humanos . ",
22
- input_labels="phrases",
23
- output_labels="forecast")
24
-
25
-
26
-
27
- iface.launch()
28
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  import openai
3
+ import csv
4
+ import os
5
+
6
+ openai.api_key = os.getenv("OPENAI_API_KEY")
7
+
8
+ datos_ingresados = [] # Lista para almacenar los datos ingresados por el usuario
9
+
10
+ def guardar_en_csv(dato):
11
+ ruta_archivo = 'datos_usuarios.csv' # Ruta relativa al archivo CSV
12
+
13
+ with open(ruta_archivo, mode='a+', newline='') as archivo_csv:
14
+ escritor_csv = csv.writer(archivo_csv, delimiter=',')
15
+ escritor_csv.writerow([dato])
16
+
17
+ print("Los datos se han guardado en el archivo CSV exitosamente.")
18
 
19
  def generate_completion(user_prompt):
20
  hidden_context = " "
 
22
  response = openai.Completion.create(
23
  model="davinci:ft-topwow-llc:rrhh-2023-09-26-22-30-27",
24
  prompt=prompt,
25
+ max_tokens=10,
26
  temperature=0,
27
  stop=["_END"]
28
  )
29
 
30
+ guardar_en_csv(user_prompt)
31
  return response.choices[0].text.strip()
 
 
 
 
 
 
 
 
 
 
 
32
 
33
+ iface = gr.Interface(
34
+ fn=generate_completion,
35
+ inputs=gr.inputs.Textbox(lines=5, placeholder='Escribe aquí la frase o las palabras '),
36
+ outputs='text',
37
+ title="Detector de potenciales renuncias",
38
+ description="Plataforma que detecta potenciales síntomas de esquizofrenia. Escriba la frase y le suma la flecha -> mas dos espacios, ejemplo: Veo monstruos en la noche -> La plataforma está basada en redes neuronales transformer usando la técnica de transfer learning y el proceso de fine tunning",
39
+ input_labels="Frase",
40
+ output_labels="Predicción"
41
+ )
42
+
43
+ iface.launch()