Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import torch | |
| # Load the model | |
| model = torch.load("path/to/basilkr_luke_ST_low_model.pth") | |
| model.eval() # Set the model to evaluation mode | |
| # Define a function to make predictions | |
| def predict(text): | |
| # Perform inference using the model | |
| # You need to adapt this based on the specific requirements of your model | |
| result = model(text) | |
| return result | |
| # Create the Gradio interface | |
| iface = gr.Interface(fn=predict, inputs="text", outputs="text") | |
| iface.launch() |