File size: 339 Bytes
0382585
e5f754c
 
 
 
0382585
 
0b04dcb
0382585
 
47eeb32
e5f754c
 
4cd32d8
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from ctransformers import AutoModelForCausalLM
import gradio as gr
import random
import time



def call_chat(message,history):
    llm = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1", model_type="Mistral")
    for text in llm(message, stream=True):
        return text


gr.ChatInterface(call_chat).launch()