Spaces:

Sannidhi
/

Psychologist-chatbot

Running

Sannidhi commited on Nov 14, 2024

Commit

c3ab959

verified ·

1 Parent(s): 082aba0

Update chatbot.py

Files changed (1) hide show

chatbot.py CHANGED Viewed

@@ -15,7 +15,7 @@ memory_length = 5
 memory = ConversationBufferWindowMemory(k=memory_length, memory_key="chat_history", return_messages=True)
 model_id = "Sasmitah/llama_16bit_2"
-model2_id = "meta-llama/Llama-3.2-1B-Instruct"
 whisper_model = "openai/whisper-small"
 model1 = AutoModelForAudioClassification.from_pretrained("ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition")
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition")

 memory = ConversationBufferWindowMemory(k=memory_length, memory_key="chat_history", return_messages=True)
 model_id = "Sasmitah/llama_16bit_2"
+model2_id = "unsloth/Llama-3.2-1B-Instruct"
 whisper_model = "openai/whisper-small"
 model1 = AutoModelForAudioClassification.from_pretrained("ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition")
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition")