Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,9 +13,6 @@ from fastapi.middleware.cors import CORSMiddleware
|
|
| 13 |
from fastapi import UploadFile, File
|
| 14 |
import fitz
|
| 15 |
|
| 16 |
-
summary = "krrishsinha/legal_summariser"
|
| 17 |
-
qna = "krrishsinha/nlpques-ans"
|
| 18 |
-
clause = "krrishsinha/clausedetectionfinal"
|
| 19 |
|
| 20 |
pdf_cache = {"text": None}
|
| 21 |
|
|
@@ -36,26 +33,26 @@ def pdfopen(filepath : str) -> str:
|
|
| 36 |
|
| 37 |
def summarizer():
|
| 38 |
|
| 39 |
-
|
| 40 |
|
| 41 |
-
o = pipeline("summarization", model=
|
| 42 |
|
| 43 |
return o
|
| 44 |
|
| 45 |
def anq():
|
| 46 |
|
| 47 |
-
|
| 48 |
|
| 49 |
-
k = pipeline("question-answering", model=
|
| 50 |
|
| 51 |
return k
|
| 52 |
|
| 53 |
def clause(sen):
|
| 54 |
-
|
| 55 |
-
|
| 56 |
|
| 57 |
tokenizer = AutoTokenizer.from_pretrained(clause)
|
| 58 |
-
model = AutoModelForSequenceClassification.from_pretrained(
|
| 59 |
config = AutoConfig.from_pretrained(clause)
|
| 60 |
|
| 61 |
inputs = tokenizer(sen, return_tensors="pt", truncation=True, padding=True)
|
|
|
|
| 13 |
from fastapi import UploadFile, File
|
| 14 |
import fitz
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
pdf_cache = {"text": None}
|
| 18 |
|
|
|
|
| 33 |
|
| 34 |
def summarizer():
|
| 35 |
|
| 36 |
+
summaryp = "krrishsinha/legal_summariser"
|
| 37 |
|
| 38 |
+
o = pipeline("summarization", model= summaryp)
|
| 39 |
|
| 40 |
return o
|
| 41 |
|
| 42 |
def anq():
|
| 43 |
|
| 44 |
+
qnap = "krrishsinha/nlpques-ans"
|
| 45 |
|
| 46 |
+
k = pipeline("question-answering", model= qnap)
|
| 47 |
|
| 48 |
return k
|
| 49 |
|
| 50 |
def clause(sen):
|
| 51 |
+
|
| 52 |
+
clausep = "krrishsinha/clausedetectionfinal"
|
| 53 |
|
| 54 |
tokenizer = AutoTokenizer.from_pretrained(clause)
|
| 55 |
+
model = AutoModelForSequenceClassification.from_pretrained(clausep)
|
| 56 |
config = AutoConfig.from_pretrained(clause)
|
| 57 |
|
| 58 |
inputs = tokenizer(sen, return_tensors="pt", truncation=True, padding=True)
|