| import streamlit as st | |
| import requests | |
| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| api_key = os.getenv('HF_API_KEY') | |
| model_path = os.getenv('MODEL_PATH') | |
| 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() | |
| st.title("My AI Prediction App") | |
| text = st.text_area("Enter text to predict") | |
| if st.button("Predict"): | |
| if text: | |
| result = get_model_predictions(text) | |
| st.write("Prediction:", result) | |
| else: | |
| st.write("Please enter some text.") |