Spaces:

ullasmrnva
/

LawBerta

Build error

ullasmrnva commited on Jun 29, 2022

Commit

8e340f7

1 Parent(s): e2f20ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,6 +16,7 @@ import numpy as np
 def make_prediction(contract):
     tokenizer = AutoTokenizer.from_pretrained('roberta-base')
     contract_df = pd.DataFrame()
     contract_df = contract_df.append({'contracts': contract},
             ignore_index=True)

 def make_prediction(contract):
     tokenizer = AutoTokenizer.from_pretrained('roberta-base')
+    final_model = TFAutoModelForSequenceClassification.from_pretrained('ullasmrnva/LawBerta')
     contract_df = pd.DataFrame()
     contract_df = contract_df.append({'contracts': contract},
             ignore_index=True)