| # gradio_app.py | |
| import gradio as gr | |
| from run_pipeline import get_fact_checked | |
| def fact_check_function(text, model): | |
| # Assume the text is already read from the user input, so we don't need to open a file here | |
| out = get_fact_checked(text, mode="fast", model=model) | |
| return out["fact_checked_md"] | |
| def create_gradio_interface(): | |
| iface = gr.Interface( | |
| title="Filtir - Fact-Checking AI generated content", | |
| allow_flagging=False, | |
| fn=fact_check_function, | |
| inputs=[ | |
| gr.Textbox( | |
| lines=6, | |
| placeholder="Enter text to fact-check...", | |
| label="Input Text", | |
| max_lines=10, | |
| ), | |
| gr.Dropdown(choices=["gpt-3.5-turbo", "gpt-4-1106-preview"], label="Model"), | |
| ], | |
| outputs=gr.Markdown(label="Filtir Output"), | |
| ) | |
| return iface | |
| if __name__ == "__main__": | |
| iface = create_gradio_interface() | |
| iface.launch() | |