test_gemma / app.py
lmastm's picture
Update app.py
f5a63cf verified
raw
history blame contribute delete
405 Bytes
import streamlit as st
import requests
API_URL = "https://api-inference.huggingface.co/models/google/gemma-7b-it"
headers = {"Authorization": st.secrets["access_token"]}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.json()
text = st.text_area("Enter a sentence !!!")
if text:
output = query({
"inputs": text,
})
st.json(output)