AMP-Classifier / app.py
nonzeroexit's picture
Update app.py
14f4c95 verified
raw
history blame
378 Bytes
import gradio as gr
import joblib
# Load your trained model
model = joblib.load("SVM.joblib") # Ensure this file is in the correct directory
def predict(sequence):
prediction = model.predict([len(sequence)]) # Adjust based on your features
return "AMP" if prediction[0] else "Non-AMP"
iface = gr.Interface(fn=predict, inputs="text", outputs="text")
iface.launch()