Update main.py

main.py

@@ -1,17 +1,18 @@
 import requests
 from bs4 import BeautifulSoup
-from fastapi import FastAPI, Request, StreamingResponse
+from fastapi import FastAPI, Request
+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")
 
 
 
@@ -83,22 +84,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-v1")
+# 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}