ShebMichel commited on
Commit
a20ee81
·
verified ·
1 Parent(s): 44eee9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,7 +44,7 @@ available_models = [
44
  "hf://ShebMichel/geobot_teacher-v0",
45
  "hf://ShebMichel/geobot_teacher-v1"]
46
 
47
- geomodel_llm = keras_nlp.models.CausalLM.from_preset("hf://ShebMichel/geobot_teacher-v0")
48
  # Function to load the selected model
49
 
50
  def load_model(model_name):
@@ -54,7 +54,7 @@ def load_model(model_name):
54
 
55
 
56
  def model_launch(input,selected_model):
57
- #geomodel_llm=load_model(selected_model)
58
  template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
59
  prompt = template.format(
60
  instruction=input,
 
44
  "hf://ShebMichel/geobot_teacher-v0",
45
  "hf://ShebMichel/geobot_teacher-v1"]
46
 
47
+ # geomodel_llm = keras_nlp.models.CausalLM.from_preset("hf://ShebMichel/geobot_teacher-v0")
48
  # Function to load the selected model
49
 
50
  def load_model(model_name):
 
54
 
55
 
56
  def model_launch(input,selected_model):
57
+ geomodel_llm=load_model(selected_model)
58
  template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
59
  prompt = template.format(
60
  instruction=input,