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("Query")
t = st.slider('Temperature', 1, 200, 2)
q = {
    "inputs":inp,
    "parameters":{"temperature":float(t)*0.5}
}
data = query(q)
print(data)

st.write(data[0]['generated_text'])
a = np.ones((100,100,3),np.uint8)
a[:,:,0]=255
st.image(a)
st.image("pers.png")

import requests

API_URL = "https://api-inference.huggingface.co/models/nvidia/segformer-b5-finetuned-ade-640-640"
headers = "" #{"Authorization": "Bearer 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"}

def query(filename):
    with open(filename, "rb") as f:
        data = f.read()
    response = requests.request("POST", API_URL, headers=headers, data=data)
    return json.loads(response.content.decode("utf-8"))

output = query("pers.png")
#st.write(output)
st.write(output[0])