update

app.py

@@ -9,7 +9,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # Load the model and tokenizer with GPU optimizations
 model = AutoModelForCausalLM.from_pretrained(
     "AlanYky/phi-3.5_tweets_instruct",
-    torch_dtype=torch.float16,  # Use FP16 for GPU
+    torch_dtype=torch.float32,
     trust_remote_code=True,
 )
 
@@ -25,7 +25,7 @@ pipe = pipeline(
 
 # Define generation arguments
 generation_args = {
-    "max_new_tokens": 50,
+    "max_new_tokens": 70,
     "return_full_text": False,
     "temperature": 0.7,
     "top_k": 20,
@@ -74,20 +74,42 @@ def clean_tweet(tweet):
 
     return cleaned_tweet.strip()
 
-# Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Tweet Generator")
-    with gr.Row():
-        instruction_input = gr.Textbox(
-            label="Instruction",
-            placeholder="Enter your tweet idea (It can be a topic, hashtag, sentence, or any format)..."
+    # Add a title with the X.com logo
+    with gr.Column(align="center"):
+        gr.Markdown(
+            """
+            <div style="text-align: center;">
+                <img src="https://upload.wikimedia.org/wikipedia/commons/b/b7/X_logo.jpg" 
+                     alt="X.com Logo" width="100">
+                <h1 style="font-size: 2.5em; margin: 0;">Tweet Generator</h1>
+                <p style="font-size: 1.2em; color: gray;">
+                    Powered by <b>AlanYky/phi-3.5_tweets_instruct</b>
+                </p>
+            </div>
+            """,
+            elem_id="header"
         )
 
-    generate_button = gr.Button("Generate")
-    output_box = gr.Textbox(label="Generated Tweet", placeholder="Your tweet will appear here.")
+    # Center the input and output components
+    with gr.Column(align="center"):
+        instruction_input = gr.Textbox(
+            label="Tweet Idea",
+            placeholder="Enter your tweet idea (It can be a topic, hashtag, sentence, or any format)...",
+            lines=2,
+            elem_id="input-box"
+        )
+        generate_button = gr.Button("Generate", elem_id="generate-button")
+        output_box = gr.Textbox(
+            label="Generated Tweet",
+            placeholder="Your tweet will appear here.",
+            lines=3,
+            elem_id="output-box"
+        )
 
+    # Connect the button to the generate function
     generate_button.click(generate_tweet, inputs=instruction_input, outputs=output_box)
 
 print("Model loaded on:", next(model.parameters()).device)
 
-demo.launch()
+demo.launch()