Kr08 commited on
Commit
1a539e2
·
verified ·
1 Parent(s): c1cd1f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -48,7 +48,7 @@ def load_summarization_model():
48
  return None
49
 
50
 
51
- @spaces.GPU(duration=60)
52
  def process_audio(audio_file, translate=False):
53
  """Process audio file"""
54
  transcriber = ChunkedTranscriber(chunk_size=5, overlap=1)
@@ -84,22 +84,26 @@ def process_audio(audio_file, translate=False):
84
  # raise gr.Error(f"Processing failed: {str(e)}")
85
 
86
 
87
- @spaces.GPU(duration=60)
88
  def summarize_text(results):
89
  """Summarize text"""
90
  try:
 
91
  summarizer = load_summarization_model()
 
92
  if summarizer is None:
93
  return "Summarization model could not be loaded."
94
-
95
- summary = summarizer(''.join(item['translated'] for item in results if 'translated' in item), max_length=150, min_length=50, do_sample=False)[0]['summary_text']
 
 
96
  return summary
97
  except Exception as e:
98
  logger.error(f"Summarization failed: {str(e)}")
99
  return "Error occurred during summarization."
100
 
101
 
102
- @spaces.GPU(duration=60)
103
  def answer_question(context, question):
104
  """Answer questions about the text"""
105
  try:
 
48
  return None
49
 
50
 
51
+ @spaces.GPU(duration=120)
52
  def process_audio(audio_file, translate=False):
53
  """Process audio file"""
54
  transcriber = ChunkedTranscriber(chunk_size=5, overlap=1)
 
84
  # raise gr.Error(f"Processing failed: {str(e)}")
85
 
86
 
87
+ @spaces.GPU(duration=120)
88
  def summarize_text(results):
89
  """Summarize text"""
90
  try:
91
+
92
  summarizer = load_summarization_model()
93
+
94
  if summarizer is None:
95
  return "Summarization model could not be loaded."
96
+ logger.info("Successfully loaded summarization Model")
97
+ full_text = ''.join(item['translated'] for item in results if 'translated' in item)
98
+ logger.info(f"\n\nWorking on text:\n{full_text}")
99
+ summary = summarizer( full_text, max_length=150, min_length=50, do_sample=False)[0]['summary_text']
100
  return summary
101
  except Exception as e:
102
  logger.error(f"Summarization failed: {str(e)}")
103
  return "Error occurred during summarization."
104
 
105
 
106
+ @spaces.GPU(duration=120)
107
  def answer_question(context, question):
108
  """Answer questions about the text"""
109
  try: