Update app.py

app.py

@@ -18,7 +18,7 @@ def compare_sentences(sentence1, sentence2):
 def generate_gpt2_response(question):
     input_ids = tokenizer.encode(question, return_tensors='pt').to(model.device)
     if input_ids.size(1) == 0:
-        return "Generated response is empty."
+        return "Generated response is empty OR Input your question"
     generated_output = model.generate(input_ids, max_length=len(input_ids[0]) + 100,
                                       num_beams=5,
                                       no_repeat_ngram_size=2,
@@ -63,8 +63,8 @@ if __name__ == '__main__':
     while True:
         
         user_input = st.text_area("Enter your question:")
-    # if user_input.lower() == 'exit':
-    #     break
+        if user_input.lower() == 'exit':
+            break
     response = find_question_and_answer(json_file_path, user_input)
     st.write(response)