Spaces:

ghudeihed
/

text_generation

Sleeping

git.name commited on Mar 5, 2024

Commit

bff9622

1 Parent(s): c0ad4b9

updating th model and the task to question answer

Files changed (3) hide show

main.py CHANGED Viewed

@@ -8,7 +8,7 @@ from transformers import pipeline
 app = FastAPI()
 # Initialize the pipeline
-pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 # Serve static files
 app.mount("/static", StaticFiles(directory="static", html=True), name="static")
@@ -20,7 +20,7 @@ async def read_index():
 # Endpoint for text-to-text generation
 @app.get("/generate-text")
-async def generate_text(input: str):
-    output = pipe_flan(input)
     print(output)
-    return {"output": output[0]["generated_text"]}

 app = FastAPI()
 # Initialize the pipeline
+question_answerer = pipeline("question-answering", model='distilbert-base-cased-distilled-squad')
 # Serve static files
 app.mount("/static", StaticFiles(directory="static", html=True), name="static")
 # Endpoint for text-to-text generation
 @app.get("/generate-text")
+async def generate_text(question: str, context: str):
+    output = question_answerer(question=question, context=context)
     print(output)
+    return {"output": f"Answer: '{output['answer']}', score: {round(output['score'], 4)}, start: {output['start']}, end: {output['end']}"}

static/index.html CHANGED Viewed

@@ -8,19 +8,22 @@
 <body>
 <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" id="textGenForm">
         <label for="text-gen-input">Text prompt:</label>
         <input
-          id="text-gen-input"
           name="text"
           type="text"
           placeholder="Ask me anything"

 <body>
 <main>
     <section id="text-gen">
+      <h2>Text generation using GPT2</h2>
       <p>
         Model:
         <a
+          href="https://huggingface.co/distilbert/distilbert-base-cased-distilled-squad"
           rel="noreferrer"
+          target="_blank">distilbert-base-cased-distilled-squad
         </a>
       </p>
       <form class="text-gen-form" id="textGenForm">
+        <label for="text-gen-context">Text context:</label>
+        <textarea id="text-gen-context" ></textarea>
+        <br />
         <label for="text-gen-input">Text prompt:</label>
         <input
+          id="text-gen-question"
           name="text"
           type="text"
           placeholder="Ask me anything"

static/script.js CHANGED Viewed

@@ -1,9 +1,9 @@
 document.addEventListener("DOMContentLoaded", () => {
   const textGenForm = document.querySelector(".text-gen-form");
-  const translateText = async (text) => {
       try {
-          const response = await fetch(`generate-text?input=${encodeURIComponent(text)}`);
           if (!response.ok) {
               throw new Error('Network response was not ok.');
           }
@@ -17,13 +17,14 @@ document.addEventListener("DOMContentLoaded", () => {
   textGenForm.addEventListener("submit", async (event) => {
       event.preventDefault();
-      const textGenInput = document.getElementById("text-gen-input");
       const textGenParagraph = document.querySelector(".text-gen-output");
       // Display a loading message or similar feedback
-      textGenParagraph.textContent = "Generating text...";
-      const output = await translateText(textGenInput.value);
       textGenParagraph.textContent = output;
   });
 });

 document.addEventListener("DOMContentLoaded", () => {
   const textGenForm = document.querySelector(".text-gen-form");
+  const translateText = async (question, context) => {
       try {
+          const response = await fetch(`generate-text?question=${encodeURIComponent(question)}&context=${encodeURIComponent(context)}`);
           if (!response.ok) {
               throw new Error('Network response was not ok.');
           }
   textGenForm.addEventListener("submit", async (event) => {
       event.preventDefault();
+      const textGenContext = document.getElementById("text-gen-context");
+      const textGenQuestion = document.getElementById("text-gen-question");
       const textGenParagraph = document.querySelector(".text-gen-output");
       // Display a loading message or similar feedback
+      textGenParagraph.textContent = "Generating answer...";
+      const output = await translateText(textGenQuestion.value, textGenContext.value);
       textGenParagraph.textContent = output;
   });
 });