Spaces:
Runtime error
Runtime error
import streamlit as st | |
import os | |
import requests | |
# Set your model and project details | |
project_name = 'my_autotrain_llm' | |
model_name = 'abhishek/llama-2-7b-hf-small-shards' | |
hf_token = 'YOUR_HUGGING_FACE_TOKEN' # Replace with your Hugging Face token | |
# Set Hugging Face API endpoint | |
hf_api_endpoint = 'https://huggingface.co/am-nandeesh/jql' | |
st.title("Streamlit App for Hugging Face Spaces") | |
user_input = st.text_area("Enter Text:", "Type your text here...") | |
if st.button("Run Model"): | |
headers = {"Authorization": f"Bearer {hf_token}"} | |
data = {"inputs": user_input} | |
response = requests.post( | |
f"{hf_api_endpoint}{model_name}/tasks/text-generation", | |
headers=headers, | |
json=data | |
) | |
if response.status_code == 200: | |
output_text = response.json()['predictions'][0] | |
st.success(f"Model Output: {output_text}") | |
else: | |
st.error(f"Error running the model. Status code: {response.status_code}") | |