Update app.py

app.py

@@ -45,7 +45,7 @@ def model_launch(input):
         instruction=input,                                                                   
         response="",
     )
-    out = gemma_lm.generate(prompt, max_length=1024)
+    out = geomodel_llm.generate(prompt, max_length=1024)
     ind = out.index('Response') + len('Response')+2
     return out[ind:]
 
@@ -135,8 +135,16 @@ def set_question_to_edit(selection):
 
 def submit_question(question_content):
     # Simulate processing by an LLM model
-    model_output = f"Model response to the question: {question_content[:100]}..."  # Only showing part for brevity
-    return model_output
+    #model_output = f"Model response to the question: {question_content[:100]}..."  # Only showing part for brevity
+    template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
+    prompt = template.format(
+        instruction=question_content,                                                                   
+        response="",
+    )
+    out = geomodel_llm.generate(prompt, max_length=1024)
+    ind = out.index('Response') + len('Response')+2
+    #return out[ind:]
+    return model_output [ind:] #model_output
 
 def clear_output():
     return "", ""