Spaces:
Paused
Paused
Commit
·
290d660
1
Parent(s):
dfa2f23
Initial Docker Space
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ class ChatBot:
|
|
| 28 |
self.model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0").to(self.device)
|
| 29 |
# Parler-TTS Setup
|
| 30 |
self.tts_model = ParlerTTSForConditionalGeneration.from_pretrained("doublesizebed/parler-tts-mini-malay").to(self.device)
|
| 31 |
-
self.tts_tokenizer = AutoTokenizer.from_pretrained("
|
| 32 |
self.description_tokenizer = AutoTokenizer.from_pretrained(self.tts_model.config.text_encoder._name_or_path)
|
| 33 |
|
| 34 |
async def get_response(self, user_input, gender):
|
|
|
|
| 28 |
self.model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0").to(self.device)
|
| 29 |
# Parler-TTS Setup
|
| 30 |
self.tts_model = ParlerTTSForConditionalGeneration.from_pretrained("doublesizebed/parler-tts-mini-malay").to(self.device)
|
| 31 |
+
self.tts_tokenizer = AutoTokenizer.from_pretrained("doublesizebed/parler-tts-mini-malay")
|
| 32 |
self.description_tokenizer = AutoTokenizer.from_pretrained(self.tts_model.config.text_encoder._name_or_path)
|
| 33 |
|
| 34 |
async def get_response(self, user_input, gender):
|