| import gradio as gr | |
| from inference import EnsembleInference | |
| model = EnsembleInference("Final_Model.pt") | |
| def predict_text(text): | |
| result = model.predict(text) | |
| return f"{result['prediction']} (Confidence: {result['confidence']})\n\nDetails:\n- RoBERTa: {result['details']['roberta_confidence']}\n- DeBERTa: {result['details']['deberta_confidence']}" | |
| iface = gr.Interface( | |
| fn=predict_text, | |
| inputs=gr.Textbox(lines=5, placeholder="Enter your text here..."), | |
| outputs="text", | |
| title="AI Detector", | |
| description="This model detects whether the text is AI-generated or human-written using an ensemble of RoBERTa and DeBERTa." | |
| ) | |
| iface.launch() | |