Update app.py

app.py

@@ -31,12 +31,18 @@ temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.0,.1)
 
 if user_input:
     for checkpoint in checkpoints:
+        
+        model_name = checkpoint.split('/')[1]
+
+        if 'flan' in model_name.lower():
+            user_input = 'generate a question: '+user_input
+        
         output = query(checkpoint,{
             "inputs": user_input,
             "temperature":temperature,
             "wait_for_model":True})
 
-        model_name = checkpoint.split('/')[1]
+        
         st.write(f'Model {model_name}: {output}')