Spaces:
Sleeping
Sleeping
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() |