codaitechs / app.py
Medtty's picture
Create app.py
ef998ea verified
raw
history blame
302 Bytes
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()