update

app.py

@@ -6,14 +6,17 @@ from peft import PeftModel, PeftConfig
 
 # Hugging Face login
 token = os.environ.get("token")
-login(token)
-print("Login is successful")
+if token:
+    login(token)
+    print("Login is successful")
+else:
+    print("Token not found. Please set your token in the environment variables.")
 
 # Model and tokenizer setup
 MODEL_NAME = "google/flan-t5-base"
-tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME, token=token)
+tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME, use_auth_token=token)
 config = PeftConfig.from_pretrained("Komal-patra/results")
-base_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+base_model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
 
 # Text generation function
@@ -50,6 +53,14 @@ span.md.svelte-8tpqd2.chatbot.prose p {
 .gradio-container {
     background: #1c1c1c; /* Dark background */
     color: white; /* Light text color */
+    height: 100vh; /* Full viewport height */
+    display: flex;
+    flex-direction: column;
+}
+/* Chatbot messages container */
+.svelte-1s78gfg {
+    flex-grow: 1; /* Allow it to grow and take available space */
+    overflow-y: auto; /* Enable vertical scrolling */
 }
 /* RED (Hex: #DB1616) for action buttons and links only */
 .clear-btn {
@@ -94,9 +105,11 @@ with gr.Blocks(css=custom_css) as demo:
             submit_button = gr.Button("Submit", elem_classes="submit-btn")
             clear = gr.Button("Clear", elem_classes="clear-btn")
 
+            # Function to handle user input
             def user(user_message, history):
                 return "", history + [[user_message, None]]
 
+            # Function to handle bot response
             def bot(history):
                 if len(history) == 1:  # Check if it's the first interaction
                     bot_message = "Hi there! How can I help you today?"
@@ -116,4 +129,4 @@ with gr.Blocks(css=custom_css) as demo:
             )
             clear.click(lambda: None, None, chatbot, queue=False)
 
-demo.launch()
+demo.launch(share=True)