Spaces:

nchen909
/

tryremote

Sleeping

nchen909 commited on Mar 22

Commit

9496d4d

•

1 Parent(s): 5eae1c6

Upload 7 files

Files changed (7) hide show

Dockerfile ADDED Viewed

+# Use the official Python 3.9 image
+FROM python:3.9
+# Set the working directory to /code
+WORKDIR /code
+# Copy the current directory contents into the container at /code
+COPY ./requirements.txt /code/requirements.txt
+# Install requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

README.md CHANGED Viewed

@@ -1,8 +1,8 @@
 ---
-title: Tryremote
 emoji: 🐢
-colorFrom: blue
-colorTo: pink
 sdk: docker
 pinned: false
 ---

 ---
+title: Fastapi T5
 emoji: 🐢
+colorFrom: purple
+colorTo: blue
 sdk: docker
 pinned: false
 ---

main.py ADDED Viewed

+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+from transformers import pipeline
+app = FastAPI()
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+@app.get("/infer_t5")
+def t5(input):
+    output = pipe_flan(input)
+    return {"output": output[0]["generated_text"]}
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+@app.get("/")
+def index() -> FileResponse:
+    return FileResponse(path="/app/static/index.html", media_type="text/html")

requirements.txt ADDED Viewed

+fastapi==0.74.*
+requests==2.27.*
+sentencepiece==0.1.*
+torch==1.11.*
+transformers==4.*
+uvicorn[standard]==0.17.*

static/index.html ADDED Viewed

+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Fast API 🤗 Space served with Uvicorn</title>
+    <link rel="stylesheet" href="style.css" />
+    <script type="module" src="script.js"></script>
+  </head>
+  <body>
+    <main>
+      <section id="text-gen">
+        <h1>Text generation using Flan T5</h1>
+        <p>
+          Model:
+          <a
+              href="https://huggingface.co/google/flan-t5-small"
+              rel="noreferrer"
+              target="_blank"
+            >google/flan-t5-small</a
+          >
+        </p>
+        <form class="text-gen-form">
+          <label for="text-gen-input">Text prompt</label>
+          <input
+            id="text-gen-input"
+            type="text"
+            value="English: Translate There are many ducks. German:"
+          />
+          <button id="text-gen-submit">Submit</button>
+          <p class="text-gen-output"></p>
+        </form>
+      </section>
+    </main>
+  </body>
+</html>

static/script.js ADDED Viewed

+const textGenForm = document.querySelector('.text-gen-form');
+const translateText = async (text) => {
+    const inferResponse = await fetch(`infer_t5?input=${text}`);
+    const inferJson = await inferResponse.json();
+    return inferJson.output;
+};
+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);
+  }
+});

static/style.css ADDED Viewed

+body {
+  --text: hsl(0 0% 15%);
+  padding: 2.5rem;
+  font-family: sans-serif;
+  color: var(--text);
+}
+body.dark-theme {
+  --text: hsl(0 0% 90%);
+  background-color: hsl(223 39% 7%);
+}
+main {
+  max-width: 80rem;
+  text-align: center;
+}
+section {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+}
+a {
+  color: var(--text);
+}
+form {
+  width: 30rem;
+  margin: 0 auto;
+}
+input {
+  width: 100%;
+}
+button {
+  cursor: pointer;
+}
+.text-gen-output {
+  min-height: 1.2rem;
+  margin: 1rem;
+  border: 0.5px solid grey;
+}