Update app.py

app.py

@@ -7,13 +7,13 @@ import random
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
-model = GPT2LMHeadModel.from_pretrained("RandomNameAnd6/DharGPT-Small")
+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)
+        real_index_state = gr.State(0)  # Initialize real_index_state properly
 
         score_display = gr.Markdown("## Real or AI - Dhar Mann\n**Current Score: 0**")
 
@@ -87,9 +87,10 @@ 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
+        # Set initial content for option boxes and real_index_state
         option1_box.value = option1
         option2_box.value = option2
+        real_index_state.set(real_index)  # Set the initial real_index_state
 
     return demo