Update app.py

app.py

@@ -296,8 +296,8 @@ 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))
+    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,"",""])
     except: