Update app.py

app.py

@@ -44,43 +44,20 @@ available_models = [
     "hf://ShebMichel/geobot_teacher-v0",
     "hf://ShebMichel/geobot_teacher-v1"]
 
-# Create a combobox for model selection
-# model_selector = widgets.Combobox(
-#     options=available_models,
-#     placeholder="Select a model",
-#     description="Model:",
-#     ensure_option=True
-# )
-# display(model_selector)
-
-
-# Dropdown widget instead of Combobox for better compatibility
-model_selector = widgets.Dropdown(
-    options=available_models,
-    description="Model:"
-)
-display(model_selector)
 
 # Function to load the selected model
-def load_model():
-    selected_model = model_selector.value
-    if selected_model:
-        geomodel_llm = keras_nlp.models.CausalLM.from_preset(selected_model)
-        return geomodel_llm
-    else:
-        print("Please select a model.")
-        return None
-
+def load_model(model_name):
+    # tokenizer = AutoTokenizer.from_pretrained(model_name)
+    # model = AutoModelForCausalLM.from_pretrained(model_name)
+    # generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
+    geomodel_llm = keras_nlp.models.CausalLM.from_preset(model_name)
+    return geomodel_llm
 
 
 
 #geomodel_llm = keras_nlp.models.CausalLM.from_preset("hf://ShebMichel/geobot_teacher-v0")
 
 def model_launch(input):
-    geomodel_llm = load_model()
-    if geomodel_llm is None:
-        return "No model selected."
-
     template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
     prompt = template.format(
         instruction=input,                                                                   
@@ -236,6 +213,13 @@ with gr.Blocks() as iface:
     
     # Submit button - Passes selected question to the model
     with gr.Row():
+        with gr.Row():
+            with gr.Column(scale=2):
+                model_dropdown = gr.Dropdown(
+                    choices=available_models,
+                    label="Select Model",
+                    value=None
+                )
         submit_button = gr.Button("Submit", interactive=False)
         clear_button = gr.Button("Clear", interactive=False)