| 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:braintumor-2023-08-30-01-15-51", |
| 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=' Insert the arrow -> after the description of the symptom, for example "I have a bad headache at night ->"'), |
| outputs='text', |
| title="Brain tumor", |
| description="Platform to detect potential brain tumor symptoms, does not replace medical professionals. ", |
| input_labels="Phrases", |
| output_labels="symptoms") |
| iface.launch() |
