clover / app.py
berak's picture
Update app.py
5d76d29
raw
history blame
617 Bytes
import streamlit as st
import pandas as pd
import numpy as np
import cv2
st.write(cv2.__version__)
print(cv2.__version__)
import json
import requests
API_URL = "https://api-inference.huggingface.co/models/gpt2"
headers = {"Authorization": f"Bearer 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"}
def query(payload):
data = json.dumps(payload)
response = requests.request("POST", API_URL, headers=headers, data=data)
return json.loads(response.content.decode("utf-8"))
inp = st.text_input("Can you please let us know more details about your ")
data = query(inp)
print(data)
st.write(data[0]['generated_text'])