Update main.py

main.py

@@ -5,14 +5,14 @@ from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 import re
 import os
-# import transformers
-# import torch
+import transformers
+import torch
 
 app = FastAPI()
 
 
 
-#summarizer = transformers.pipeline("summarization", model="Falconsai/text_summarization")
+summarizer = transformers.pipeline("summarization", model="Falconsai/text_summarization")
 
 
 
@@ -54,7 +54,7 @@ async def root(item: Item):
         json={'prompt': "{ 'User': 'Summarize the following news article: '" + article + "}",
         "temperature":0.6,
         "topP":0.9,
-        "maxTokens": 200}, stream=False)
+        "maxTokens": 200}, stream=True)
 
         response_content = response.content.decode('utf-8')
 
@@ -113,22 +113,22 @@ async def root(item: Item):
         return {"error": str(e), "status_code": 500}
     
 
-# @app.post("/summarize-v1")
-# async def root(item: Item):
+@app.post("/summarize-v3")
+async def root(item: Item):
     
-#     try:
+    try:
 
-#         article = extract_article_content(item.url)
+        article = extract_article_content(item.url)
 
-#         if len(article) == 0:
-#             return {'summary': ""}
+        if len(article) == 0:
+            return {'summary': ""}
 
-#         summ = summarizer(article, max_length=1000, min_length=30, do_sample=False)
+        summ = summarizer(article, max_length=1000, min_length=30, do_sample=False)
  
-#         return {"summary":summ}
+        return {"summary":summ}
         
-#     except requests.RequestException as e:
-#         return {"error": str(e), "status_code": 500}
+    except requests.RequestException as e:
+        return {"error": str(e), "status_code": 500}