krrishsinha commited on
Commit
7598eba
·
verified ·
1 Parent(s): d4bb30d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -10
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
- summarypath = r"E:/FastAPI/Lawlytics/legal_summariser"
40
 
41
- o = pipeline("summarization", model= summary)
42
 
43
  return o
44
 
45
  def anq():
46
 
47
- qnapath = r"E:/FastAPI/Lawlytics/nlpques&ans"
48
 
49
- k = pipeline("question-answering", model= qna)
50
 
51
  return k
52
 
53
  def clause(sen):
54
-
55
- clausepath = r"E:/FastAPI/Lawlytics/clausedetectionfinal"
56
 
57
  tokenizer = AutoTokenizer.from_pretrained(clause)
58
- model = AutoModelForSequenceClassification.from_pretrained(clause)
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)