Spaces:
Sleeping
Sleeping
import gradio as gr | |
import torch | |
# Load the model | |
model = torch.load("chatbot_model.pt") # Adjust as needed | |
# Define the chatbot function | |
def chatbot(input_text): | |
# Use your model inference function here | |
response = get_response(input_text) # Call to your response function | |
return response | |
# Define the Gradio interface | |
iface = gr.Interface( | |
fn=chatbot, # The function that will be called | |
inputs="text", # User will input text | |
outputs="text" # Model will output text | |
) | |
# Launch the app | |
iface.launch() | |