Update app.py

app.py

@@ -9,12 +9,6 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')
 model = GPT2LMHeadModel.from_pretrained('RandomNameAnd6/DharGPT-Small')
 
-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
-
 # Read real titles from file
 with open('dhar_mann_titles.txt', 'r') as file:
     dhar_mann_titles = file.readlines()
@@ -86,6 +80,7 @@ def generate_ai_title():
             title = title.strip()  # Remove any extra whitespace characters like newlines
             if string_similarity_index(generated_title, title):
                 is_unique = False
+                print(f"Regenerating! Original title was: {generated_title}")
                 break
 
         if is_unique: