Update generate_transcript.py

generate_transcript.py

@@ -6,7 +6,6 @@ import transformers
 import pickle
 from tqdm import tqdm
 import warnings
-import spaces
 from groq import Groq
 import os
 
@@ -18,7 +17,7 @@ class TranscriptProcessor:
     A class to generate and rewrite podcast-style transcripts using a specified language model.
     """
 
-    def __init__(self, text_file_path, model_name="meta-llama/Llama-3.1-8B-Instruct"):
+    def __init__(self, text_file_path,transcript_output_path,tts_output_path, model_name="llama3-70b-8192"):
         """
         Initialize with the path to the cleaned text file and the model name.
         
@@ -27,16 +26,11 @@ class TranscriptProcessor:
             model_name (str): Name of the language model to use.
         """
         self.text_file_path = text_file_path
-        self.transcript_output_path = './resources/data.pkl'
-        self.tts_output_path = './resources/podcast_ready_data.pkl'
+        self.transcript_output_path = transcript_output_path
+        self.tts_output_path = tts_output_path
         self.model_name = model_name
-        self.accelerator = Accelerator()
-        self.model = transformers.pipeline(
-            "text-generation",
-            model=self.model_name,
-            model_kwargs={"torch_dtype": torch.bfloat16},
-            device_map="auto"
-        )
+
+       
         self.transcript_prompt = """
         You are a world-class podcast writer, working as a ghost writer for top podcast hosts.
         You will write the dialogue with engaging interruptions, anecdotes, and curiosity-led questions.