Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import torch
+import random
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 def load_model(model_path, dtype):
@@ -42,6 +43,7 @@ def generate(
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
 
     if seed_checkbox:
+        seed = random.randint(1, 100000)
         torch.manual_seed(seed)
 
     outputs = model.generate(
@@ -107,6 +109,7 @@ additional_inputs = [
         value=False,
         label="Use Random Seed",
         info="Check to use a random seed for the generation process",
+        change=update_ui,
     ),
     gr.Number(
         value=42,
@@ -122,6 +125,13 @@ additional_inputs = [
     ),
 ]
 
+def update_ui(seed_checkbox):
+    if seed_checkbox:
+        return gr.Number.update(visible=False)
+    else:
+        return gr.Number.update(visible=True)
+
+
 examples = [
     [
         "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.",