Alberto Carmona commited on
Commit
80d9102
·
1 Parent(s): 057f5f4

Stop loading the model until space upgrade with GPU

Browse files
Files changed (1) hide show
  1. functions.py +4 -1
functions.py CHANGED
@@ -22,6 +22,9 @@ def extract_text(url: str):
22
 
23
 
24
  def summarize_text(text: str):
 
 
 
25
  print(['summarize_text', 'start'])
26
  input_text = f'<s>Instruction: Elabora un resume del siguiente texto.\nInput: {text}\nOutput: '
27
  batch = tokenizer(input_text, return_tensors='pt')
@@ -50,4 +53,4 @@ def load_model(peft_model_id):
50
  return model, tokenizer
51
 
52
 
53
- model, tokenizer = load_model("milyiyo/opt-6.7b-lora-sag-t3000-v300-v2")
 
22
 
23
 
24
  def summarize_text(text: str):
25
+ # Pending to load the model first
26
+ return ''
27
+
28
  print(['summarize_text', 'start'])
29
  input_text = f'<s>Instruction: Elabora un resume del siguiente texto.\nInput: {text}\nOutput: '
30
  batch = tokenizer(input_text, return_tensors='pt')
 
53
  return model, tokenizer
54
 
55
 
56
+ # model, tokenizer = load_model("milyiyo/opt-6.7b-lora-sag-t3000-v300-v2")