First commit

app.py

@@ -0,0 +1,25 @@
+import streamlit as st
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+
+# Instantiate the model
+model = AutoModelForSeq2SeqLM.from_pretrained("dantedgp/question-generator")
+tokenizer = AutoTokenizer.from_pretrained("dantedgp/question-generator")
+
+def generate_response(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)
+    question = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    print('question: ', question)
+    return question
+
+st.set_page_config(page_title='AI Quiz Generator')
+st.title('AI Quiz Generator')
+
+# Text input
+result = []
+txt_input = st.text_area('Enter your text', '', height=200)
+
+if len(result):
+    st.info(result)

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch