sooolee commited on
Commit
92f9c99
·
1 Parent(s): ee65db2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
40
  model = AutoModelForSeq2SeqLM.from_pretrained(config.base_model_name_or_path, device_map='auto') # load_in_8bit=True,
41
  model = PeftModel.from_pretrained(model, peft_model_id, device_map='auto')
42
 
43
- def summarize(path):
44
  transcript = load_data(file_obj)
45
  texts = preprocessing(transcript)
46
  inputs = tokenizer(texts, return_tensors="pt", padding=True, )
 
40
  model = AutoModelForSeq2SeqLM.from_pretrained(config.base_model_name_or_path, device_map='auto') # load_in_8bit=True,
41
  model = PeftModel.from_pretrained(model, peft_model_id, device_map='auto')
42
 
43
+ def summarize(file_obj):
44
  transcript = load_data(file_obj)
45
  texts = preprocessing(transcript)
46
  inputs = tokenizer(texts, return_tensors="pt", padding=True, )