Update app.py

app.py

@@ -1,20 +1,13 @@
-from fastapi import FastAPI
+import gradio as gr
 import joblib
-import numpy as np
-
-app = FastAPI()
 
 # Load your trained model
 model = joblib.load("SVM.joblib")  # Ensure this file is in the correct directory
 
-@app.get("/")
-def home():
-    return {"message": "AMP Classifier API is running!"}
+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")
 
-@app.post("/predict")
-def predict(sequence: str):
-    # Convert sequence to features (modify if needed)
-    features = np.array([len(sequence)])  # Example: length of sequence
-    prediction = model.predict([features])
-    
-    return {"sequence": sequence, "is_AMP": bool(prediction[0])}
+iface.launch()