Update app.py

app.py

@@ -10,10 +10,10 @@ tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
 model = GPT2LMHeadModel.from_pretrained("gpt2-medium")
 
 def generate_text(prompt):
-    input_ids = tokenizer.encode(prompt, return_tensors="pt")
-    output = model.generate(input_ids, max_length=48, temperature=0.85, do_sample=True)
-    text = tokenizer.decode(output[0], skip_special_tokens=True)
-    return text
+  input_ids = tokenizer.encode(prompt, return_tensors="pt")
+  output = model.generate(input_ids, max_length=48, temperature=0.85, do_sample=True)
+  text = tokenizer.decode(output[0], skip_special_tokens=True)
+  return text
 
 # Read real titles from file
 with open('dhar_mann_titles.txt', 'r') as file:
@@ -21,7 +21,7 @@ with open('dhar_mann_titles.txt', 'r') as file:
 
 # Function to generate an AI title (dummy implementation)
 def generate_ai_title():
-    inputs = tokenizer(["<|startoftext|>"]*1, return_tensors="pt")
+    inputs = tokenizer(["<|startoftext|>"]*1, return_tensors = "pt")
     outputs = model.generate(**inputs, max_new_tokens=50, use_cache=True, temperature=0.85, do_sample=True)
     return (tokenizer.batch_decode(outputs)[0])[15:-13]
 
@@ -48,7 +48,7 @@ def update_options():
 def create_interface():
     with gr.Blocks() as demo:
         score = gr.State(0)
-        real_index_state = gr.State(0)  # Initialize real_index_state properly
+        real_index_state = gr.State(0)
 
         score_display = gr.Markdown("## Real or AI - Dhar Mann\n**Current Score: 0**")
 
@@ -87,10 +87,9 @@ def create_interface():
         continue_button.click(on_continue, inputs=score, outputs=[option1_box, option2_box, real_index_state, score_display, choice, result_text, continue_button, restart_button])
         restart_button.click(on_restart, outputs=[option1_box, option2_box, real_index_state, score_display, choice, result_text, continue_button, restart_button])
 
-        # Set initial content for option boxes and real_index_state
+        # Set initial content for option boxes
         option1_box.value = option1
         option2_box.value = option2
-        real_index_state.set(real_index)  # Set the initial real_index_state
 
     return demo