inoid commited on
Commit
a3a731c
1 Parent(s): b802fa4

Use environement variables with os.environ function

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. spanish_medica_llm.py +2 -0
app.py CHANGED
@@ -77,6 +77,7 @@ def train_model(*inputs):
77
  # return [gr.update(visible=True, value=["diffusers_model.zip"]), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)]
78
  run_training_process()
79
 
 
80
  return f"Train Model Sucessful!!!"
81
 
82
  def stop_model(*input):
 
77
  # return [gr.update(visible=True, value=["diffusers_model.zip"]), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)]
78
  run_training_process()
79
 
80
+
81
  return f"Train Model Sucessful!!!"
82
 
83
  def stop_model(*input):
spanish_medica_llm.py CHANGED
@@ -329,6 +329,8 @@ HUB_MODEL_ID = 'somosnlp/spanish_medica_llm'
329
  MAX_TRAINING_STEPS = int(1500/2)
330
  MAX_TRAINING_STEPS = 2
331
 
 
 
332
  def loadSpanishTokenizer():
333
  """
334
 
 
329
  MAX_TRAINING_STEPS = int(1500/2)
330
  MAX_TRAINING_STEPS = 2
331
 
332
+ TOKEN_NAME = TOKEN_MISTRAL_NAME
333
+
334
  def loadSpanishTokenizer():
335
  """
336