Eduarr commited on
Commit
afc0dfc
·
1 Parent(s): e8e575f

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +44 -0
app.py ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = " "
21
+ prompt = hidden_context + user_prompt
22
+ response = openai.Completion.create(
23
+ model="davinci:ft-topwow-llc:esquizofrenia-2023-08-31-20-33-36",
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 síntomas de esquizofrenia",
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()
44
+