Spaces:

klasocki
/

comma-fixer

Sleeping

klasocki commited on Aug 19, 2023

Commit

a9ea03f

•

1 Parent(s): 4a35568

Add a home page for the huggingface hub

Breaks integration tests,
and now you cannot access the API endpoint from outside (is it good?
I guess it is)

Files changed (6) hide show

Dockerfile CHANGED Viewed

@@ -9,7 +9,7 @@ ENV PATH="/venv/bin:$PATH"
 COPY requirements.txt .
 COPY test-requirements.txt .
 RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
 COPY src/baseline.py src/baseline.py
 ENV TRANSFORMERS_CACHE=/coma-fixer/.cache

 COPY requirements.txt .
 COPY test-requirements.txt .
 RUN pip install --upgrade pip
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
 COPY src/baseline.py src/baseline.py
 ENV TRANSFORMERS_CACHE=/coma-fixer/.cache

README.md CHANGED Viewed

@@ -12,4 +12,6 @@ app_port: 8000
 `docker log [id]` for logs from the container.
-`docker build -t comma-fixer --target test .` for tests

 `docker log [id]` for logs from the container.
+`docker build -t comma-fixer --target test .` for tests
+`git push hub` to deploy to huggingface hub, after adding a remote

app.py CHANGED Viewed

@@ -1,23 +1,18 @@
 import uvicorn
 from fastapi import FastAPI, HTTPException
 from src.baseline import BaselineCommaFixer
 import logging
 logger = logging.Logger(__name__)
 logging.basicConfig(level=logging.INFO)
-app = FastAPI() #TODO router?
 logger.info('Loading the baseline model...')
 app.baseline_model = BaselineCommaFixer()
-@app.get('/')
-async def root():
-    return ("Welcome to the comma fixer. Send a POST request to /fix-commas or /baseline/fix-commas with a string "
-            "'s' in the JSON body to try "
-            "out the functionality.")
 @app.post('/baseline/fix-commas/')
 async def fix_commas_with_baseline(data: dict):
     json_field_name = 's'
@@ -30,6 +25,13 @@ async def fix_commas_with_baseline(data: dict):
         raise HTTPException(status_code=400, detail=msg)
-if __name__ == '__main__':
-    uvicorn.run("app:app", reload=True, port=8000)

 import uvicorn
 from fastapi import FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 from src.baseline import BaselineCommaFixer
 import logging
 logger = logging.Logger(__name__)
 logging.basicConfig(level=logging.INFO)
+app = FastAPI()  # TODO router?
 logger.info('Loading the baseline model...')
 app.baseline_model = BaselineCommaFixer()
 @app.post('/baseline/fix-commas/')
 async def fix_commas_with_baseline(data: dict):
     json_field_name = 's'
         raise HTTPException(status_code=400, detail=msg)
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+@app.get('/')
+async def index() -> FileResponse:
+    return FileResponse(path="static/index.html", media_type="text/html")
+if __name__ == '__main__':
+    uvicorn.run("app:app", port=8000)

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="comma-fixing">
+        <h2>Fixing commas using Transformers</h2>
+        <p>
+          Model:
+          <a
+            href="https://huggingface.co/oliverguhr/fullstop-punctuation-multilang-large"
+            rel="noreferrer"
+            target="_blank"
+            >oliverguhr/fullstop-punctuation-multilang-large
+          </a>
+        </p>
+        <form class="comma-fixing-form">
+          <label for="comma-fixing-input">Text with incorrect commas</label>
+          <input
+            id="comma-fixing-input"
+            type="text"
+            value="This is however a very bad, and terrible sentence grammatically that is."
+          />
+          <button id="comma-fixing-submit">Submit</button>
+          <p class="comma-fixing-output"></p>
+        </form>
+      </section>
+    </main>
+  </body>
+</html>

static/script.js ADDED Viewed

+const commaFixingForm = document.querySelector(".comma-fixing-form");
+const fixCommas = async (text) => {
+    const inferResponse = await fetch(`baseline/fix-commas/`, {
+        method: "POST",
+        body: JSON.stringify({
+            s: text
+        }),
+        headers: {
+            "Content-type": "application/json; charset=UTF-8"
+        }
+    });
+    const inferJson = await inferResponse.json();
+    return inferJson.s;
+};
+commaFixingForm.addEventListener("submit", async (event) => {
+    event.preventDefault();
+    const commaFixingInput = document.getElementById("comma-fixing-input");
+    const commaFixingParagraph = document.querySelector(".comma-fixing-output");
+    commaFixingParagraph.textContent = await fixCommas(commaFixingInput.value);
+});

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;
+}