Add dropdown for model type

app.py

@@ -6,10 +6,25 @@ import click
 from rasa.nlu.model import Interpreter
 
 
-MODEL_PATH = "woz_nlu_agent/models/nlu"
+RASA_MODEL_PATH = "woz_nlu_agent/models/nlu"
 interpreter = None
 
-def predict(input):
+MODEL_TYPES = {
+    "Out-of-scope classifier": "oos",
+    "Intent classifier": "intent_transformer",
+    "Intent and Entity extractor": "rasa_intent_entity"
+}
+
+def predict(model_type, input):
+    if MODEL_TYPES[model_type] == "rasa_intent_entity":
+        return rasa_predict(input)
+    elif MODEL_TYPES[model_tyoe] == "oos":
+        return "TODO: out of scope"
+    elif MODEL_TYPES[model_type] == "intent_transformer":
+        return "TODO:: intent_transformer"
+
+
+def rasa_predict(input):
 
     def rasa_output(text):
         message = str(text).strip()
@@ -45,7 +60,7 @@ def main():
     
     interpreter = Interpreter.load(MODEL_PATH)
     print("Model loaded.")
-    iface = gr.Interface(fn=predict, inputs="text", outputs="text")
+    iface = gr.Interface(fn=predict, inputs=[gr.Dropdown(MODEL_TYPES.keys()), "text"], outputs="text")
     iface.launch()