ensure required args

app.py

@@ -17,20 +17,23 @@
 from argparse import Namespace
 args = Namespace()
 
-arg_dict = {  
-    "run_gradio": True,
-    "model_name_or_path": "facebook/opt-125m",
-    # "model_name_or_path": "facebook/opt-1.3b",
-    # "model_name_or_path": "facebook/opt-2.7b",
-    "max_new_tokens": 200,
-    "use_sampling": True,
-    "sampling_temp": 0.7,
-    "use_gpu": True,
-    "seeding_scheme": "markov_1",
-    "gamma": 0.25,
-    "delta": 2.0, 
-    "normalizers": "",
-    "ignore_repeated_bigrams": False,
+arg_dict = {
+    'run_gradio': True, 
+    'demo_public': False, 
+    'model_name_or_path': 'facebook/opt-2.7b', 
+    'prompt_max_length': None, 
+    'max_new_tokens': 200, 
+    'generation_seed': 123, 
+    'use_sampling': True, 
+    'sampling_temp': 0.7, 
+    'use_gpu': True, 
+    'seeding_scheme': 'markov_1', 
+    'gamma': 0.25, 
+    'delta': 2.0, 
+    'normalizers': '', 
+    'ignore_repeated_bigrams': False, 
+    'detection_z_threshold': 4.0, 
+    'select_green_tokens': True
 }
 
 args.__dict__.update(arg_dict)