sitammeur commited on
Commit
0446166
·
verified ·
1 Parent(s): 258b0b5

Update src/app/response.py

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