inoid commited on
Commit
496d856
1 Parent(s): ea4e88a

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 -2
app.py CHANGED
@@ -39,6 +39,7 @@ def evaluate_model():
39
  #image = pipe(prompt).images[0]
40
  return(f"Evaluate Model {os.environ.get('HF_LLM_MODEL_ID')} from dataset {os.environ.get('HF_LLM_DATASET_ID')}")
41
 
 
42
  def train_model(*inputs):
43
  if "IS_SHARED_UI" in os.environ:
44
  raise gr.Error("This Space only works in duplicated instances")
 
39
  #image = pipe(prompt).images[0]
40
  return(f"Evaluate Model {os.environ.get('HF_LLM_MODEL_ID')} from dataset {os.environ.get('HF_LLM_DATASET_ID')}")
41
 
42
+
43
  def train_model(*inputs):
44
  if "IS_SHARED_UI" in os.environ:
45
  raise gr.Error("This Space only works in duplicated instances")
spanish_medica_llm.py CHANGED
@@ -539,11 +539,11 @@ def configAndRunTraining(basemodel, dataset, eval_dataset, tokenizer):
539
 
540
  trainer.push_to_hub()
541
 
542
-
543
 
544
  def run_training_process():
545
  #Loggin to Huggin Face
546
- login(token = os.environ('HG_FACE_TOKEN'))
547
  tokenizer = loadSpanishTokenizer()
548
  medicalSpanishDataset = loadSpanishDataset()
549
  train_dataset, eval_dataset, test_dataset = splitDatasetInTestValid(
 
539
 
540
  trainer.push_to_hub()
541
 
542
+
543
 
544
  def run_training_process():
545
  #Loggin to Huggin Face
546
+ login(token = os.environ.get('HG_FACE_TOKEN'))
547
  tokenizer = loadSpanishTokenizer()
548
  medicalSpanishDataset = loadSpanishDataset()
549
  train_dataset, eval_dataset, test_dataset = splitDatasetInTestValid(