Update app.py

app.py

@@ -14,9 +14,10 @@ model.to('cpu')
 
 # Define the function that generates text from a prompt
 def generate_text(prompt, temperature):
+    word_count = len(prompt.split())
+    if word_count < 5:
+        return "Please provide at least 5 words in the prompt."
 
-    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')
 
@@ -51,4 +52,4 @@ interface = gr.Interface(
     description="TopicGPT processes the input and returns a reasonably accurate guess of the topic/theme of a given conversation."
 )
 
-interface.launch()
+interface.launch()