Kr08 commited on
Commit
a806a3e
·
verified ·
1 Parent(s): 1aa0b18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -88,7 +88,7 @@ def load_summarization_model():
88
  return None
89
 
90
 
91
- @spaces.GPU(duration=120)
92
  def process_audio(audio_file, translate=False):
93
  """Process audio file"""
94
  transcriber = ChunkedTranscriber(chunk_size=5, overlap=1)
@@ -124,26 +124,26 @@ def process_audio(audio_file, translate=False):
124
  # raise gr.Error(f"Processing failed: {str(e)}")
125
 
126
 
127
- @spaces.GPU(duration=120)
128
- def summarize_text(text):
129
- """Summarize text"""
130
- try:
131
 
132
- summarizer = load_summarization_model()
133
 
134
- if summarizer is None:
135
- return "Summarization model could not be loaded."
136
- logger.info("Successfully loaded summarization Model")
137
- # logger.info(f"\n\n {text}\n")
138
 
139
- summary = summarizer(text, max_length=150, min_length=50, do_sample=False)[0]['summary_text']
140
- return summary
141
- except Exception as e:
142
- logger.error(f"Summarization failed: {str(e)}")
143
- return "Error occurred during summarization."
144
 
145
 
146
- @spaces.GPU(duration=120)
147
  def answer_question(context, question):
148
  """Answer questions about the text"""
149
  try:
 
88
  return None
89
 
90
 
91
+ @spaces.GPU(duration=180)
92
  def process_audio(audio_file, translate=False):
93
  """Process audio file"""
94
  transcriber = ChunkedTranscriber(chunk_size=5, overlap=1)
 
124
  # raise gr.Error(f"Processing failed: {str(e)}")
125
 
126
 
127
+ @spaces.GPU(duration=180)
128
+ # def summarize_text(text):
129
+ # """Summarize text"""
130
+ # try:
131
 
132
+ # summarizer = load_summarization_model()
133
 
134
+ # if summarizer is None:
135
+ # return "Summarization model could not be loaded."
136
+ # logger.info("Successfully loaded summarization Model")
137
+ # # logger.info(f"\n\n {text}\n")
138
 
139
+ # summary = summarizer(text, max_length=150, min_length=50, do_sample=False)[0]['summary_text']
140
+ # return summary
141
+ # except Exception as e:
142
+ # logger.error(f"Summarization failed: {str(e)}")
143
+ # return "Error occurred during summarization."
144
 
145
 
146
+ @spaces.GPU(duration=180)
147
  def answer_question(context, question):
148
  """Answer questions about the text"""
149
  try: