Spaces:
Runtime error
Runtime error
import numpy as np | |
import pickle | |
from keras.preprocessing.text import Tokenizer | |
from keras.preprocessing.sequence import pad_sequences | |
from sklearn.preprocessing import LabelEncoder | |
from keras.models import load_model | |
import gradio as gr | |
# Load model and encoders | |
model = load_model('doctor_ai_model.h5') | |
with open('tokenizer.pkl', 'rb') as f: | |
tokenizer = pickle.load(f) | |
with open('label_encoder.pkl', 'rb') as f: | |
label_encoder = pickle.load(f) | |
# Function to get response from model | |
def get_response(input_text): | |
# Preprocess input text | |
input_sequences = tokenizer.texts_to_sequences([input_text]) | |
input_tensor = pad_sequences(input_sequences, maxlen=100) # Adjust maxlen as necessary | |
# Make prediction | |
predicted_label = model.predict(input_tensor) | |
decoded_label = label_encoder.inverse_transform([np.argmax(predicted_label)]) | |
return decoded_label[0] | |
# Create Gradio interface | |
iface = gr.Interface(fn=get_response, inputs="text", outputs="text", title="Doctor AI", | |
description="Ask your health-related questions and get advice!") | |
# Launch the interface | |
iface.launch() | |