Update app.py

app.py

@@ -37,7 +37,7 @@ rewardmodel_id = "nicholasKluge/RewardModelPT"
 toxicitymodel_id = "nicholasKluge/ToxicityModelPT"
 
 # specify the device (cuda if available)
-device = torch.device("cpu") # "cuda" if torch.cuda.is_available() else "cpu"
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
 
 # load the models (chatbot, reward model, toxicity model)
 model = AutoModelForCausalLM.from_pretrained(model_id)
@@ -100,7 +100,7 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
     gr.Markdown("""<h1><center>Aira Demo (Portuguese) 🤓💬</h1></center>""")
     gr.Markdown(intro)
 
-    chatbot = gr.Chatbot(label="Aira", height=500)
+    chatbot = gr.Chatbot(label="Aira").style(height=500)
     msg = gr.Textbox(label="Escreva uma pergunta ou instrução para Aira ...", placeholder="Olá Aira, como vai você?")
 
     # Parameters to control the generation