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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -54,15 +54,16 @@ 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,
61
- response="",
62
- )
63
- out = geomodel_llm.generate(prompt, max_length=1024)
64
- ind = out.index('Response') + len('Response')+2
65
- return out[ind:]
 
66
 
67
  # Global variable to store loaded JSON data
68
  loaded_data = {}
 
54
 
55
 
56
  def model_launch(input,selected_model):
57
+ return f"Model: {selected_model}\nResponse to: {input}"
58
+ # geomodel_llm=load_model(selected_model)
59
+ # template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
60
+ # prompt = template.format(
61
+ # instruction=input,
62
+ # response="",
63
+ # )
64
+ # out = geomodel_llm.generate(prompt, max_length=1024)
65
+ # ind = out.index('Response') + len('Response')+2
66
+ # return out[ind:]
67
 
68
  # Global variable to store loaded JSON data
69
  loaded_data = {}