Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ token = os.getenv("HF_TOKEN")
|
|
| 9 |
login(token=token)
|
| 10 |
|
| 11 |
# Whisper Model Optimization
|
| 12 |
-
model = "openai/whisper-
|
| 13 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 14 |
|
| 15 |
processor = AutoProcessor.from_pretrained(model)
|
|
@@ -24,7 +24,7 @@ transcriber = pipeline(
|
|
| 24 |
)
|
| 25 |
|
| 26 |
# LLaMA Model Optimization
|
| 27 |
-
LLAMA = "meta-llama/
|
| 28 |
llama_quant_config = BitsAndBytesConfig(
|
| 29 |
load_in_4bit=True,
|
| 30 |
bnb_4bit_use_double_quant=True,
|
|
@@ -47,7 +47,8 @@ def process_audio(audio_file):
|
|
| 47 |
|
| 48 |
# Transcribe audio
|
| 49 |
transcript = transcriber(audio_file)["text"]
|
| 50 |
-
|
|
|
|
| 51 |
# Generate meeting minutes
|
| 52 |
system_message = "You are an assistant that produces minutes of meetings from transcripts, with summary, key discussion points, takeaways and action items with owners, in markdown."
|
| 53 |
user_prompt = f"Below is an extract transcript of a Denver council meeting. Please write minutes in markdown, including a summary with attendees, location and date; discussion points; takeaways; and action items with owners.\n{transcript}"
|
|
|
|
| 9 |
login(token=token)
|
| 10 |
|
| 11 |
# Whisper Model Optimization
|
| 12 |
+
model = "openai/whisper-tiny"
|
| 13 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 14 |
|
| 15 |
processor = AutoProcessor.from_pretrained(model)
|
|
|
|
| 24 |
)
|
| 25 |
|
| 26 |
# LLaMA Model Optimization
|
| 27 |
+
LLAMA = "meta-llama/Llama-3.2-3B-Instruct"
|
| 28 |
llama_quant_config = BitsAndBytesConfig(
|
| 29 |
load_in_4bit=True,
|
| 30 |
bnb_4bit_use_double_quant=True,
|
|
|
|
| 47 |
|
| 48 |
# Transcribe audio
|
| 49 |
transcript = transcriber(audio_file)["text"]
|
| 50 |
+
del transcriber
|
| 51 |
+
del processor
|
| 52 |
# Generate meeting minutes
|
| 53 |
system_message = "You are an assistant that produces minutes of meetings from transcripts, with summary, key discussion points, takeaways and action items with owners, in markdown."
|
| 54 |
user_prompt = f"Below is an extract transcript of a Denver council meeting. Please write minutes in markdown, including a summary with attendees, location and date; discussion points; takeaways; and action items with owners.\n{transcript}"
|