Update app.py

app.py

@@ -1,30 +1,25 @@
+import streamlit as st
+import requests
 import os
 from dotenv import load_dotenv
-from flask import Flask, request, jsonify
-import requests
 
 load_dotenv()
 api_key = os.getenv('HF_API_KEY')
 model_path = os.getenv('MODEL_PATH')
 
-app = Flask(__name__)
-
 def get_model_predictions(text):
     headers = {"Authorization": f"Bearer {api_key}"}
     payload = {"inputs": text}
     response = requests.post(f"https://api.huggingface.co/models/{model_path}", headers=headers, json=payload)
     return response.json()
 
-@app.route('/predict', methods=['POST'])
-def predict():
-    data = request.get_json()
-    text = data['text']
-    prediction = get_model_predictions(text)
-    return jsonify(prediction)
+st.title("My AI Prediction App")
 
-@app.route('/', methods=['GET'])
-def index():
-    return "Welcome to My AI!"
+text = st.text_area("Enter text to predict")
 
-if __name__ == '__main__':
-    app.run(debug=True)
+if st.button("Predict"):
+    if text:
+        result = get_model_predictions(text)
+        st.write("Prediction:", result)
+    else:
+        st.write("Please enter some text.")