Update app.py

app.py

@@ -43,19 +43,16 @@ def infer(prompt, samples, steps, scale):
     return images_list
 
 
-def generate_movie(state):
-    state["title"] = None
-    random_number = randrange(data.shape[0])
-    plot = data.iloc[random_number]["plot_synopsis_sum"]
+def generate_movie():
+    seed = randrange(data.shape[0])
+    plot = data.iloc[seed]["plot_synopsis_sum"]
     image = infer(plot, 1, 50, 7.5)
-    state["number"] = random_number
-    state["title"] = data.iloc[int(random_number)]["title"]
 
-    return image["sample"][0], state
+    return image["sample"][0], seed
 
 
-def movie_title(state):
-    return state["title"]
+def movie_title(seed):
+    return data.iloc[int(seed)]["title"]
 
 
 css = """
@@ -181,9 +178,9 @@ with gr.Blocks(css=css) as demo:
                 label="Generated images", show_label=False, elem_id="image"
             ).style(height="auto")
 
-            state = gr.State({"number": None, "title": None})
+            seed = gr.Number(visible=False)
 
-            b1.click(generate_movie, inputs=state, outputs=[image, state])
-            b2.click(movie_title, inputs=state, outputs=text)
+            b1.click(generate_movie, inputs=None, outputs=[image, seed])
+            b2.click(movie_title, inputs=seed, outputs=text)
 
 demo.launch()