Spaces:
Runtime error
Runtime error
File size: 1,137 Bytes
04cbe77 838ec20 b694fa5 4b8add8 b694fa5 838ec20 8ff6443 b694fa5 398a30c b694fa5 838ec20 19cb6e6 b694fa5 0440734 838ec20 19cb6e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
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()
|