update

app.py

@@ -6,9 +6,11 @@ from peft import PeftModel, PeftConfig
 
 # Hugging Face login
 token = os.environ.get("token")
-login(token)
-print("login is succesful")
-max_length=512
+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"
@@ -18,7 +20,7 @@ base_model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
 
 # Text generation function
-def generate_text(prompt, max_length=512):
+def generate_text(prompt, max_length=150):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         input_ids=inputs["input_ids"],
@@ -30,20 +32,20 @@ def generate_text(prompt, max_length=512):
     return generated_text
 
 # Custom CSS for the UI
-background_image_path = 'https://www.shlegal-technology.com/sites/default/files/insight/ExploringTheLegislativeBackgroundBANNER.jpg'
+background_image_path = '/mnt/data/image.png'
 custom_css = f"""
 .message.pending {{
-    background: #A8C4D6;
+    background: rgba(168, 196, 214, 0.5);
 }}
 /* Response message */
 .message.bot.svelte-1s78gfg.message-bubble-border {{
-    border-color: #266B99;
-    background-color: rgba(255, 255, 255, 0.7); /* Make the message box slightly transparent */
+    border-color: rgba(38, 107, 153, 0.5);
+    background: rgba(255, 255, 255, 0.1);
 }}
 /* User message */
 .message.user.svelte-1s78gfg.message-bubble-border {{
-    background: #9DDDF9;
-    border-color: #9DDDF9;
+    background: rgba(157, 221, 249, 0.5);
+    border-color: rgba(157, 221, 249, 0.5);
 }}   
 /* For both user and response message as per the document */
 span.md.svelte-8tpqd2.chatbot.prose p {{
@@ -51,12 +53,12 @@ span.md.svelte-8tpqd2.chatbot.prose p {{
 }}
 /* Chatbot container */
 .gradio-container {{
-    background: #1c1c1c; /* Dark background */
     color: white; /* Light text color */
     background-image: url('{background_image_path}'); /* Add background image */
     background-size: cover; /* Cover the entire container */
     background-position: center; /* Center the image */
     background-repeat: no-repeat; /* Do not repeat the image */
+    background-color: rgba(0, 0, 0, 0.5); /* Semi-transparent dark background */
 }}
 /* RED (Hex: #DB1616) for action buttons and links only */
 .clear-btn {{
@@ -124,4 +126,4 @@ with gr.Blocks(css=custom_css) as demo:
     )
     clear.click(lambda: None, None, chatbot, queue=False)
 
-demo.launch()
+demo.launch()