Update app.py

app.py

@@ -1,10 +1,8 @@
 import spaces
 import gradio as gr
-
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-
 class ModelProcessor:
     def __init__(self, repo_id="HuggingFaceTB/cosmo-1b"):
         self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -20,7 +18,6 @@ class ModelProcessor:
         # Set padding token as end-of-sequence token
         self.tokenizer.pad_token = self.tokenizer.eos_token
 
-    
     @torch.inference_mode()
     def process_data_and_compute_statistics(self, prompt):
         # Tokenize the prompt and move to the device
@@ -64,7 +61,6 @@ class ModelProcessor:
 
         return 1 if (ranks.clamp(max=4) * nll.clamp(max=4)).mean() < 5.2 else 0
 
-
 processor = ModelProcessor()
 
 @spaces.GPU(duration=180)
@@ -75,7 +71,6 @@ def detect(prompt):
     else:
         return "The text is likely **not generated** by a language model."
 
-
 with gr.Blocks(
     css="""
     .gradio-container {
@@ -89,14 +84,21 @@ with gr.Blocks(
     }
     .gr-button {
         background-color: #007bff;
-        color: white;        padding: 10px 20px;
+        color: white;        
+        padding: 10px 20px;
         border-radius: 4px;
     }
     .gr-button:hover {
-        background-color:    }
+        background-color: #0056b3;
+    }
     .hyperlinks a {
         margin-right: 10px;
     }
+    .output-text {
+        text-align: center;
+        font-size: 24px;
+        font-weight: bold;
+    }
 """
 ) as demo:
     with gr.Row():
@@ -116,7 +118,7 @@ with gr.Blocks(
                 label="Prompt",
             )
             submit_btn = gr.Button("Submit", variant="primary")
-            output = gr.Markdown()
+            output = gr.Markdown(elem_id="output-text")
 
     submit_btn.click(fn=detect, inputs=prompt, outputs=output)