Update app.py

app.py

@@ -52,19 +52,17 @@ def load_model(model_name):
     geomodel_llm = keras_nlp.models.CausalLM.from_preset(str(model_name))
     return geomodel_llm
 
-
-
 def model_launch(input):
-    return f"Model: {first_model}\n\nResponse to: {input}"
-    # geomodel_llm=load_model(first_model)
-    # template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
-    # prompt = template.format(
-    #     instruction=input,                                                                   
-    #     response="",
-    # )
-    # out = geomodel_llm.generate(prompt, max_length=1024)
-    # ind = out.index('Response') + len('Response')+2
-    # return out[ind:]
+    #return f"Model: {first_model}\n\nResponse to: {input}"
+    geomodel_llm=load_model(first_model)
+    template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
+    prompt = template.format(
+        instruction=input,                                                                   
+        response="",
+    )
+    out = geomodel_llm.generate(prompt, max_length=1024)
+    ind = out.index('Response') + len('Response')+2
+    return out[ind:]
 
 # Global variable to store loaded JSON data
 loaded_data = {}