import gradio as gr
import torch

# Load your model (replace with your model loading code)
model = torch.load('your_model.pth')

def predict(input_data):
    # Process input and make prediction
    return model(input_data)

iface = gr.Interface(fn=predict, inputs="text", outputs="text")
iface.launch()