Updated ask model

This commit is contained in:
Atridad Lahiji 2024-01-12 17:49:30 -07:00
parent 7a9e1da0e2
commit 17213829f0
No known key found for this signature in database

View file

@ -26,7 +26,7 @@ func ReplicateTextGeneration(prompt string) (string, error) {
Events: []replicate.WebhookEventType{"start", "completed"},
}
prediction, predictionError := client.Run(context.Background(), "mistralai/mixtral-8x7b-instruct-v0.1:7b3212fbaf88310cfef07a061ce94224e82efc8403c26fc67e8f6c065de51f21", input, &webhook)
prediction, predictionError := client.Run(context.Background(), "mistralai/mistral-7b-instruct-v0.2:79052a3adbba8116ebc6697dcba67ad0d58feff23e7aeb2f103fc9aa545f9269", input, &webhook)
if predictionError != nil {
return "", predictionError