Update app.py

app.py

@@ -22,7 +22,7 @@ def generate(
     temperature,
     top_p,
     top_k,
-    seed_checkbox,
+    random_seed,
     seed,
     model_path="roborovski/superprompt-v1",
     dtype="fp16",
@@ -42,10 +42,12 @@ def generate(
     input_text = f"{prompt}, {history}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
 
-    if seed_checkbox:
+    if random_seed:
         seed = random.randint(1, 100000)
         torch.manual_seed(seed)
-
+    else:
+        torch.manual_seed(seed)
+        
     outputs = model.generate(
         input_ids,
         max_new_tokens=max_new_tokens,
@@ -121,7 +123,7 @@ additional_inputs = [
         choices=["fp32", "fp16"],
         value="fp16",
         label="Model Precision",
-        info="Select the precision of the model: fp32 or fp16",
+        info="fp32 is more precised, fp16 is faster and less memory consuming",
     ),
 ]
 
@@ -130,11 +132,11 @@ examples = [
     [
         "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.",
         None,
-        None,
-        None,
-        None,
-        None,
-        None,
+        512,
+        1.2,
+        0.5,
+        1,
+        50,
         False,
         None,
         "roborovski/superprompt-v1",