sitammeur commited on
Commit
bbce3fa
·
verified ·
1 Parent(s): f560283

Update src/app/response.py

Browse files
Files changed (1) hide show
  1. src/app/response.py +2 -2
src/app/response.py CHANGED
@@ -13,13 +13,13 @@ from src.config import (
13
  stream,
14
  repetition_penalty,
15
  )
16
- from src.app.model import load_model_and_tokenizer
17
  from src.logger import logging
18
  from src.exception import CustomExceptionHandling
19
 
20
 
21
  # Model, tokenizer and processor
22
- model, tokenizer, processor = load_model_and_tokenizer(model_name, device)
23
 
24
 
25
  @spaces.GPU(duration=120)
 
13
  stream,
14
  repetition_penalty,
15
  )
16
+ from src.app.model import load_model_tokenizer_and_processor
17
  from src.logger import logging
18
  from src.exception import CustomExceptionHandling
19
 
20
 
21
  # Model, tokenizer and processor
22
+ model, tokenizer, processor = load_model_tokenizer_and_processor(model_name, device)
23
 
24
 
25
  @spaces.GPU(duration=120)