RinInori commited on
Commit
aa56743
1 Parent(s): ac7945c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -96,8 +96,10 @@ g = gr.Interface(
96
  fn=evaluate,
97
  inputs=gr.inputs.Textbox(label="Input", placeholder="Type your input here..."),
98
  outputs=gr.outputs.Textbox(label="Output"),
99
- title="Text Generation",
100
- description="Enter an input, and the model will generate a response based on the input.",
 
 
101
  theme="default",
102
  )
103
 
 
96
  fn=evaluate,
97
  inputs=gr.inputs.Textbox(label="Input", placeholder="Type your input here..."),
98
  outputs=gr.outputs.Textbox(label="Output"),
99
+ title="ChatBot and Text Generation",
100
+ description="Enter an input, and the model will generate a response based on the input." \
101
+ BASE_MODEL = "TheBloke/vicuna-7B-1.1-HF" \
102
+ LORA_WEIGHTS = "RinInori/vicuna_finetuned_6_sentiments" #Fine-tuned Alpaca model for sentiment analysis \,
103
  theme="default",
104
  )
105