diff --git a/lib/replicate.go b/lib/replicate.go index e251791..cf46095 100644 --- a/lib/replicate.go +++ b/lib/replicate.go @@ -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