Spaces:

numind
/

NuExtract

Running on L4

Alexandre-Numind commited on Sep 10

Commit

b1b293d

•

1 Parent(s): 9b07b7c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -296,7 +296,7 @@ model.eval()
 def get_prediction(text,template,example):
     size = len(tokenizer(text)["input_ids"])
     print(size)
-    if size > 2000:
         raise gr.Error("Max token for input text is 2000 tokes. Yours is: "+str(size))
     try:
         prompt = create_prompt(text,template,[example,"",""])

 def get_prediction(text,template,example):
     size = len(tokenizer(text)["input_ids"])
     print(size)
+    if size > 5000:
         raise gr.Error("Max token for input text is 2000 tokes. Yours is: "+str(size))
     try:
         prompt = create_prompt(text,template,[example,"",""])