Update app.py

app.py

@@ -5,7 +5,8 @@ import gradio as gr
 from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer
 from transformers import Trainer
 
-BASE_MODEL = "TheBloke/vicuna-7B-1.1-HF"
+#BASE_MODEL = "TheBloke/vicuna-7B-1.1-HF"
+BASE_MODEL = "RinInori/vicuna_finetuned_6_sentiments"
 
 model = LlamaForCausalLM.from_pretrained(
     BASE_MODEL,
@@ -50,6 +51,6 @@ iface = gr.Interface(
     inputs="text", 
     outputs="text",
     title="Chatbot",
-    description="This vicuna app is using this model: https://huggingface.co/TheBloke/vicuna-7B-1.1-HF"
+    description="This vicuna app is using this model: https://huggingface.co/RinInori/vicuna_finetuned_6_sentiments"
 )
 iface.launch()