Upload app.py

app.py

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