Update app.py

app.py

@@ -26,6 +26,10 @@ def load_model(model_name):
         tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b")
         model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")
         return model, tokenizer
+    if model_name == "Qwen":
+        tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-72B-Instruct")
+        model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-72B-Instruct")
+        return model, tokenizer
     else:
         st.error(f"Model {model_name} not available.")
         return None, None
@@ -41,7 +45,7 @@ def main():
     st.title("Question Generation From Given Text")
     context = st.text_area("Enter text","Laughter is the best medicine.")
     st.write("Select a model and provide the text to generate questions.")
-    model_choice = st.selectbox("Select a model", ["T5", "Llama3", "Llama3-Instruct","Phi3","Gemma"])
+    model_choice = st.selectbox("Select a model", ["T5", "Llama3", "Llama3-Instruct","Phi3","Gemma","Qwen"])
 
     if st.button("Generate Questions"):
         model, tokenizer = load_model(model_choice)