inoid commited on
Commit
b0e3b0d
1 Parent(s): 5fdc8a9

Fix generate process

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. spanish_medica_llm.py +1 -0
app.py CHANGED
@@ -62,7 +62,6 @@ def finnetuning_model(*inputs):
62
  def stop_model(*input):
63
  return f"Model with Gradio!"
64
 
65
-
66
  with gr.Blocks() as demo:
67
  gr.Markdown("Start typing below and then click **Run** to see the output.")
68
  with gr.Row():
 
62
  def stop_model(*input):
63
  return f"Model with Gradio!"
64
 
 
65
  with gr.Blocks() as demo:
66
  gr.Markdown("Start typing below and then click **Run** to see the output.")
67
  with gr.Row():
spanish_medica_llm.py CHANGED
@@ -727,6 +727,7 @@ def run_finnetuning_process():
727
  print (train_dataset[5])
728
  configAndRunFineTuning(base_model,train_dataset, eval_dataset, tokenizer)
729
 
 
730
  def generate_response(query):
731
  max_new_tokens=256
732
  temperature=0.1
 
727
  print (train_dataset[5])
728
  configAndRunFineTuning(base_model,train_dataset, eval_dataset, tokenizer)
729
 
730
+
731
  def generate_response(query):
732
  max_new_tokens=256
733
  temperature=0.1