Small changes

app.py

@@ -7,40 +7,14 @@ class GradioInference():
     self.model = pipeline(model="humeur/lab2_id2223")
     self.yt = None
   
-  def __call__(self, link, subs):
+  def __call__(self, link):
     if self.yt is None:
       self.yt = YouTube(link)
     path = self.yt.streams.filter(only_audio=True)[0].download(filename="tmp.mp4")
 
     results = self.model(path)
-
-    if subs == ".srt":
-      return self.srt(results["segments"])
-    elif subs == ".csv":
-      return self.csv(results["segments"])
     return results["text"]
 
-  def srt(self, segments):
-    output = ""
-    for i, segment in enumerate(segments):
-      output += f"{i+1}\n"
-      output += f"{self.format_time(segment['start'])} --> {self.format_time(segment['end'])}\n"
-      output += f"{segment['text']}\n\n"
-    return output
-
-  def csv(self, segments):
-    output = ""
-    for segment in segments:
-      output += f"{segment['start']},{segment['end']},{segment['text']}\n"
-    return output
-
-  def format_time(self, time):
-    hours = time//3600
-    minutes = (time - hours*3600)//60
-    seconds = time - hours*3600 - minutes*60
-    milliseconds = (time - int(time))*1000
-    return f"{int(hours):02d}:{int(minutes):02d}:{int(seconds):02d},{int(milliseconds):03d}"
-
   def populate_metadata(self, link):
     self.yt = YouTube(link)
     return self.yt.thumbnail_url, self.yt.title
@@ -65,8 +39,6 @@ with block:
     )
     with gr.Group():
         with gr.Box():
-          with gr.Row().style(equal_height=True):
-            wt = gr.Radio(["No", ".srt", ".csv"], label="Result with timestamps?")
           link = gr.Textbox(label="YouTube Link")
           title = gr.Label(label="Video Title")
           with gr.Row().style(equal_height=True):
@@ -74,6 +46,6 @@ with block:
             text = gr.Textbox(label="Transcription", placeholder="Transcription Output", lines=10)
           with gr.Row().style(equal_height=True):
               btn = gr.Button("Transcribe")
-          btn.click(gio, inputs=[link, wt], outputs=[text])
+          btn.click(gio, inputs=[link], outputs=[text])
           link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
 block.launch()