tg / app.py
Eduarr's picture
Create app.py
683ec4a
import gradio as gr
import openai
import os
openai.api_key = os.getenv("OPENAI_API_KEY")
def generate_completion(user_prompt):
hidden_context = " "
prompt = hidden_context + user_prompt
response = openai.Completion.create(
model="davinci:ft-topwow-llc:tumorgastrico-2023-09-14-20-11-13",
prompt=prompt,
max_tokens=30,
temperature=0,
stop=["_END"]
)
return response.choices[0].text.strip()
iface = gr.Interface(fn=generate_completion,
inputs=gr.inputs.Textbox(lines=5, placeholder='Escriba la frase aqui'),
outputs='text',
title="Detector de potenciales síntomas de tumor gastrico",
description="Plataforma que detecta potenciales síntomas de tumor gástrico. Escriba la frase y le suma la flecha -> mas dos espacios, ejemplo: Tengo un intenso dolor en la boca del estomago -> ",
input_labels="frase",
output_labels="prediccion")
iface.launch()