Update app.py

app.py

@@ -1,24 +1,22 @@
 from fastapi import FastAPI
 from transformers import pipeline
 
-#creating a new FastAPI app instance
-app=FastAPI()
+# Creating a new FastAPI app instance
+app = FastAPI()
 
-# calling Hugging Face Model from pipeline
-# here I am using, google/flan-t5-small (Text2Text Generating Model)
-pipe = pipeline("text2text-generation", model="google/flan-t5-small")
+# Calling the Hugging Face Model from the pipeline
+# Here I am using the Facebook/bart-large-cnn model for summarization
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-#creating routes
+# Creating routes
 @app.get("/")
 def home():
     return {"message": "Your FastAPI and Model is Running"}
 
-#define a function to handle the GET request at '/generate'
-@app.get("/generate")
-def generate(text: str):
-    # use the pipeline to generate text from given input text
-    output = pipe(text, max_new_tokens=150, temperature=0.3, top_k=30)  # Adjust these parameters as per your output accuracy
-    print(output)  # Debugging line
-    #return the correct key based on model output
-    return {"output": output[0]['generated_text']}
+@app.get("/chat")
+def chat(text: str):
+    # Use the summarizer pipeline to generate a summary from the given input text
+    output = summarizer(text, max_length=130, min_length=30, do_sample=False)
+    # Return the summary from the output
+    return {"summary": output[0]['summary_text']}