Upload 2 files

app.py

@@ -0,0 +1,41 @@
+import streamlit as st
+import json
+import requests
+import io 
+
+### Inputs
+x = st.text_input('Add your prompt here')
+
+###Load variables
+model_location = 'Mobius-labs/JenL_demo'
+
+
+### API Request
+#API_TOKEN = st.secrets["API_TOKEN"]
+API_TOKEN = "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"
+
+API_URL = "https://api-inference.huggingface.co/models/" + model_location
+
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+def query(payload):
+    data = json.dumps(payload)
+    response = requests.request("POST", API_URL, headers=headers, data=data)
+
+    return io.BytesIO(response.content)
+
+
+        
+### Button
+def process_api_output():
+    data = query(x)
+
+    st.image(data, output_format="JPEG")
+
+b = st.button('Submit prompt')
+
+if b:
+    process_api_output()
+
+#st.write(x, 'model_name is', model_name)
+#st.write(x, 'model_location is', model_location)

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+streamlit
+torch