Update app.py

app.py

@@ -14,11 +14,12 @@ model.to('cpu')
 
 # Define the function that generates text from a prompt
 def generate_text(prompt, temperature, top_p=1.0):
+
+    print(prompt)
+    
     prompt_with_eos = " #CONTEXT# " + prompt + " #TOPIC# "  # Add the string "EOS" to the end of the prompt
     input_tokens = tokenizer.encode(prompt_with_eos, return_tensors='pt')
 
-    print(prompt_with_eos)
-
     input_tokens = input_tokens.to('cpu')
 
     generated_text = prompt_with_eos  # Start with the initial prompt plus "EOS"