Initial Commit

README.md

@@ -1,12 +1,8 @@
 ---
-title: Python Http Server
-emoji: 👀
+title: Python HTTP Server
+emoji: 🐍
 colorFrom: blue
 colorTo: yellow
 sdk: gradio
-sdk_version: 2.9.0
 app_file: app.py
-pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,57 @@
+import os
+import json
+import requests
+from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
+from urllib.parse import parse_qs, urlparse
+
+from inference import t5_infer
+
+# https://huggingface.co/settings/tokens
+# https://huggingface.co/spaces/{username}/{space}/settings
+API_TOKEN = os.getenv('BIG_GAN_TOKEN')
+
+class RequestHandler(SimpleHTTPRequestHandler):
+    def do_GET(self):
+        if self.path == "/":
+            self.path = "index.html"
+
+            return SimpleHTTPRequestHandler.do_GET(self)
+
+        if self.path.startswith("/biggan_infer"):
+            input = parse_qs(urlparse(self.path).query).get("input", None)[0]
+
+            output = requests.request(
+                "POST",
+                "https://api-inference.huggingface.co/models/osanseviero/BigGAN-deep-128",
+                headers={"Authorization": f"Bearer {API_TOKEN}"},
+                data=json.dumps(input)
+            )
+
+            self.send_response(200)
+            self.send_header("Content-Type", "application/json")
+            self.end_headers()
+
+            self.wfile.write(output.content)
+
+            return SimpleHTTPRequestHandler
+
+        elif self.path.startswith("/t5_infer"):
+            input = parse_qs(urlparse(self.path).query).get("input", None)
+
+            output = t5_infer(input)
+
+            self.send_response(200)
+            self.send_header("Content-Type", "application/json")
+            self.end_headers()
+
+            self.wfile.write(json.dumps({"output": output}).encode("utf-8"))
+
+            return SimpleHTTPRequestHandler
+
+        else:
+            return SimpleHTTPRequestHandler.do_GET(self)
+
+
+server = ThreadingHTTPServer(("", 7860), RequestHandler)
+
+server.serve_forever()

index.js

@@ -0,0 +1,48 @@
+if (document.location.search.includes('dark-theme=true')) {
+  document.body.classList.add('dark-theme');
+}
+
+const textToImage = async (text) => {
+  const inferenceResponse = await fetch(`/biggan_infer?input=${text}`);
+  const inferenceBlob = await inferenceResponse.blob();
+
+  return URL.createObjectURL(inferenceBlob);
+};
+
+const translateText = async (text) => {
+  const inferResponse = await fetch(`/t5_infer?input=${text}`);
+  const inferJson = await inferResponse.json();
+
+  return inferJson.output;
+};
+
+const imageGenSelect = document.getElementById('image-gen-input');
+const imageGenImage = document.querySelector('.image-gen-output');
+const textGenForm = document.querySelector('.text-gen-form');
+
+imageGenSelect.addEventListener('change', async (event) => {
+  const value = event.target.value;
+
+  try {
+    imageGenImage.src = await textToImage(value);
+  } catch (err) {
+    console.error(err);
+  }
+});
+
+textGenForm.addEventListener('submit', async (event) => {
+  event.preventDefault();
+
+  const textGenInput = document.getElementById('text-gen-input');
+  const textGenParagraph = document.querySelector('.text-gen-output');
+
+  try {
+    textGenParagraph.textContent = await translateText(textGenInput.value);
+  } catch (err) {
+    console.error(err);
+  }
+});
+
+textToImage(imageGenSelect.value)
+  .then((image) => (imageGenImage.src = image))
+  .catch(console.error);

inference.py

@@ -0,0 +1,10 @@
+from transformers import T5Tokenizer, T5ForConditionalGeneration
+
+tokenizer = T5Tokenizer.from_pretrained("t5-small")
+model = T5ForConditionalGeneration.from_pretrained("t5-small")
+
+def t5_infer(input):
+  input_ids = tokenizer(input, return_tensors="pt").input_ids
+  outputs = model.generate(input_ids)
+
+  return tokenizer.decode(outputs[0], skip_special_tokens=True)

requirements.txt

@@ -0,0 +1 @@
+sentencepiece==0.1

style.css

@@ -0,0 +1,48 @@
+
+
+body {
+  --text: hsl(0 0% 15%);
+  font-family: sans-serif;
+  background-color: hsl(0 0% 96%);
+  color: var(--text);
+}
+body.dark-theme {
+  --text: hsl(0 0% 90%);
+  background-color: hsl(223 39% 7%);
+}
+
+main {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  max-width: 80rem;
+  text-align: center;
+}
+
+a {
+  color: var(--text);
+}
+
+select, input, button, .text-gen-output {
+  padding: 0.5rem 1rem;
+}
+
+select, img, input {
+  margin: 0.5rem auto 1rem;
+}
+
+form {
+  width: 25rem;
+  margin: 0 auto;
+}
+
+input {
+  width: 70%;
+}
+
+.text-gen-output {
+  min-height: 1rem;
+  margin: 0;
+  align-self: start;
+  border: 2px solid var(--text);
+}