Spaces:

mflash123
/

test

Sleeping

mflash123 commited on Jan 16, 2025

Commit

54783ac

1 Parent(s): 5fb0645

Add application file

Files changed (6) hide show

Dockerfile CHANGED Viewed

@@ -1,22 +1,27 @@
-# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
-# you will also find guides on how best to write your Dockerfile
 FROM python:3.9
-# The two following lines are requirements for the Dev Mode to be functional
-# Learn more about the Dev Mode at https://huggingface.co/dev-mode-explorers
-RUN useradd -m -u 1000 user
-WORKDIR /app
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
-COPY --chown=user . /app
 USER user
 ENV HOME=/home/user \
 	PATH=/home/user/.local/bin:$PATH
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

+# 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"]

main.py CHANGED Viewed

@@ -1,5 +1,20 @@
 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")

+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 CHANGED Viewed

@@ -1,2 +1,6 @@
-fastapi
-uvicorn[standard]

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

static/index.html CHANGED Viewed

@@ -1,24 +1,36 @@
-<main>
-    <section id="text-gen">
-      <h2>Text generation using Flan T5</h2>
-      <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="German: There are many ducks"
-        />
-        <button id="text-gen-submit">Submit</button>
-        <p class="text-gen-output"></p>
-      </form>
-    </section>
-  </main>

+<!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 CHANGED Viewed

@@ -1,17 +1,21 @@
-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");
-  textGenParagraph.textContent = await translateText(textGenInput.value);
 });

+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

File without changes