Jhjoon05 commited on
Commit
cf9d62f
·
1 Parent(s): 1071876

fixed model size bug

Browse files
Files changed (1) hide show
  1. modules/model_Inference.py +14 -11
modules/model_Inference.py CHANGED
@@ -8,19 +8,20 @@ from datetime import datetime
8
  class ModelInference():
9
  def __init__(self):
10
  print("\nInitializing Model..\n")
11
- self.default_model = "large-v2"
12
- self.model = whisper.load_model(self.default_model)
13
 
14
  def transcribe_file(self,fileobj
15
- ,model,lang,subformat,istranslate,
16
  progress=gr.Progress()):
17
 
18
  def progress_callback(progress_value):
19
  progress(progress_value,desc="Transcribing..")
20
 
21
- if model != self.default_model or model==None:
22
  progress(0,desc="Initializing Model..")
23
- self.model = whisper.load_model(model)
 
24
 
25
  if lang == "Automatic Detection" :
26
  lang = None
@@ -51,15 +52,16 @@ class ModelInference():
51
  return f"Done! Subtitle is in the outputs folder.\n\n{subtitle}"
52
 
53
  def transcribe_youtube(self,youtubelink
54
- ,model,lang,subformat,istranslate,
55
  progress=gr.Progress()):
56
 
57
  def progress_callback(progress_value):
58
  progress(progress_value,desc="Transcribing..")
59
 
60
- if model != self.default_model or model==None:
61
  progress(0,desc="Initializing Model..")
62
- self.model = whisper.load_model(model)
 
63
 
64
  if lang == "Automatic Detection" :
65
  lang = None
@@ -89,15 +91,16 @@ class ModelInference():
89
  return f"Done! Subtitle file is in the outputs folder.\n\n{subtitle}"
90
 
91
  def transcribe_mic(self,micaudio
92
- ,model,lang,subformat,istranslate,
93
  progress=gr.Progress()):
94
 
95
  def progress_callback(progress_value):
96
  progress(progress_value,desc="Transcribing..")
97
 
98
- if model != self.default_model or model==None:
99
  progress(0,desc="Initializing Model..")
100
- self.model = whisper.load_model(model)
 
101
 
102
  if lang == "Automatic Detection" :
103
  lang = None
 
8
  class ModelInference():
9
  def __init__(self):
10
  print("\nInitializing Model..\n")
11
+ self.current_model_size = "large-v2"
12
+ self.model = whisper.load_model(self.current_model_size)
13
 
14
  def transcribe_file(self,fileobj
15
+ ,model_size,lang,subformat,istranslate,
16
  progress=gr.Progress()):
17
 
18
  def progress_callback(progress_value):
19
  progress(progress_value,desc="Transcribing..")
20
 
21
+ if model_size != self.current_model_size:
22
  progress(0,desc="Initializing Model..")
23
+ self.current_model_size = model_size
24
+ self.model = whisper.load_model(model_size)
25
 
26
  if lang == "Automatic Detection" :
27
  lang = None
 
52
  return f"Done! Subtitle is in the outputs folder.\n\n{subtitle}"
53
 
54
  def transcribe_youtube(self,youtubelink
55
+ ,model_size,lang,subformat,istranslate,
56
  progress=gr.Progress()):
57
 
58
  def progress_callback(progress_value):
59
  progress(progress_value,desc="Transcribing..")
60
 
61
+ if model_size != self.current_model_size:
62
  progress(0,desc="Initializing Model..")
63
+ self.current_model_size = model_size
64
+ self.model = whisper.load_model(model_size)
65
 
66
  if lang == "Automatic Detection" :
67
  lang = None
 
91
  return f"Done! Subtitle file is in the outputs folder.\n\n{subtitle}"
92
 
93
  def transcribe_mic(self,micaudio
94
+ ,model_size,lang,subformat,istranslate,
95
  progress=gr.Progress()):
96
 
97
  def progress_callback(progress_value):
98
  progress(progress_value,desc="Transcribing..")
99
 
100
+ if model_size != self.current_model_size:
101
  progress(0,desc="Initializing Model..")
102
+ self.current_model_size = model_size
103
+ self.model = whisper.load_model(model_size)
104
 
105
  if lang == "Automatic Detection" :
106
  lang = None