Neuracraft / streamlit_app.py
Erfan11's picture
Create streamlit_app.py
2c3db68 verified
raw
history blame
679 Bytes
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.")