Spaces:

Mtropy
/

AI-Chat-test

Runtime error

Mtropy commited on Sep 14, 2023

Commit

26626d2

•

1 Parent(s): 8a00e40

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/Wizard-Vicuna-7B-Uncensored-GGML")
 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/vicuna-7B-v1.5-GGML")
 def chatbot(message, history):
     result = generator(message, max_length=30, num_return_sequences=1)