Spaces:

Mtropy
/

AI-Chat-test

Runtime error

Mtropy commited on Sep 14, 2023

Commit

8a00e40

•

1 Parent(s): 5aaab84

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 import transformers
 from transformers import pipeline
-generator = pipeline("text-generation", model="TheBloke/llama2_7b_chat_uncensored-GPTQ")
 def chatbot(message, history):
     result = generator(message, max_length=30, num_return_sequences=1)

 import transformers
 from transformers import pipeline
+generator = pipeline("text-generation", model="TheBloke/Wizard-Vicuna-7B-Uncensored-GGML")
 def chatbot(message, history):
     result = generator(message, max_length=30, num_return_sequences=1)