phamngoctukts commited on
Commit
8fc786c
·
verified ·
1 Parent(s): 3ceac67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -9,15 +9,11 @@ import traceback
9
  from pydub import AudioSegment
10
  import librosa
11
  from utils.vad import get_speech_timestamps, collect_chunks, VadOptions
12
- from transformers import MllamaForConditionalGeneration, AutoProcessor, TextIteratorStreamer
13
- import torch
14
- from huggingface_hub import login
15
  import os
16
  from PIL import Image
17
  from threading import Thread
18
- ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
19
- model = MllamaForConditionalGeneration.from_pretrained(ckpt,torch_dtype=torch.bfloat16).to("cpu")
20
- processor = AutoProcessor.from_pretrained(ckpt)
21
  r = sr.Recognizer()
22
 
23
  @dataclass
 
9
  from pydub import AudioSegment
10
  import librosa
11
  from utils.vad import get_speech_timestamps, collect_chunks, VadOptions
12
+ from huggingface_hub import InferenceClient
 
 
13
  import os
14
  from PIL import Image
15
  from threading import Thread
16
+ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
17
  r = sr.Recognizer()
18
 
19
  @dataclass