yashxx07 commited on
Commit
bad2aa2
·
verified ·
1 Parent(s): fa9a64f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -32
main.py CHANGED
@@ -1,12 +1,11 @@
1
  import requests
2
  from bs4 import BeautifulSoup
3
- from fastapi import FastAPI, Request
4
- from fastapi.responses import StreamingResponse
5
  from pydantic import BaseModel
6
  import re
7
- import os
8
- # import transformers
9
- # import torch
10
 
11
  class Item(BaseModel):
12
  url: str
@@ -15,14 +14,9 @@ class Item(BaseModel):
15
  app = FastAPI()
16
 
17
 
18
-
19
- #summarizer = transformers.pipeline("summarization", model="Falconsai/text_summarization")
20
-
21
-
22
-
23
  def extract_article_content(url):
24
  try:
25
- # Fetch the HTML content of the article URL
26
  response = requests.get(url)
27
 
28
  soup = BeautifulSoup(response.text, 'html.parser')
@@ -113,24 +107,3 @@ async def root(item: Item):
113
  return {"error": str(e), "status_code": 500}
114
 
115
 
116
- # @app.post("/summarize-v3")
117
- # async def root(item: Item):
118
-
119
- # try:
120
-
121
- # article = extract_article_content(item.url)
122
-
123
- # if len(article) == 0:
124
- # return {'summary': ""}
125
-
126
- # summ = summarizer(article, max_length=1000, min_length=30, do_sample=False)
127
-
128
- # return {"summary":summ}
129
-
130
- # except requests.RequestException as e:
131
- # return {"error": str(e), "status_code": 500}
132
-
133
-
134
-
135
-
136
-
 
1
  import requests
2
  from bs4 import BeautifulSoup
3
+ from fastapi import FastAPI#, Request
4
+ #from fastapi.responses import StreamingResponse
5
  from pydantic import BaseModel
6
  import re
7
+
8
+
 
9
 
10
  class Item(BaseModel):
11
  url: str
 
14
  app = FastAPI()
15
 
16
 
 
 
 
 
 
17
  def extract_article_content(url):
18
  try:
19
+
20
  response = requests.get(url)
21
 
22
  soup = BeautifulSoup(response.text, 'html.parser')
 
107
  return {"error": str(e), "status_code": 500}
108
 
109