Adding display

app.py

@@ -5,7 +5,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 model = AutoModelForSeq2SeqLM.from_pretrained("dantedgp/question-generator")
 tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-small")
 
-def generate_response(txt):
+def generate_questions(txt):
     input_text = f"ask: {txt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids
     outputs = model.generate(input_ids, max_new_tokens=100, do_sample=False)
@@ -21,5 +21,9 @@ st.title('AI Quiz Generator')
 result = []
 txt_input = st.text_area('Enter your text', '', height=200)
 
-if len(result):
-    st.info(result)
+if txt_input is not None:
+    col = st.columns(1)
+    questions = generate_questions(txt_input)
+    
+    col.header("Questions")
+    col.write(questions)