Update app.py

app.py

@@ -6,16 +6,14 @@ import pandas as pd
 import os
 import torch
 
-# Check if CUDA (GPU) is available
+# Check GPU availability (for debugging)
 print("CUDA available:", torch.cuda.is_available())
-
 if torch.cuda.is_available():
     print("GPU Name:", torch.cuda.get_device_name(0))
 else:
     print("No GPU detected. Running on CPU.")
 
-    
-# Set a seed for reproducibility
+# Set seed for reproducibility
 set_seed(42)
 
 # Define the six premium generation models:
@@ -43,20 +41,20 @@ grammar_model_names = [
     "hassaanik/grammar-correction-model"
 ]
 
-# Function to load generation pipelines, specifying GPU if available.
+# Determine device: Use GPU (0) if available, otherwise CPU (-1)
+device = 0 if torch.cuda.is_available() else -1
+
+# Function to load generation pipelines with appropriate device setting.
 def load_generation_pipeline(model_name):
     try:
-        # Use device=0 if GPU is available; otherwise, use CPU (device=-1)
-        device = 0 if torch.cuda.is_available() else -1
         return pipeline("text-generation", model=model_name, device=device)
     except Exception as e:
         print(f"Error loading generation model {model_name}: {e}")
         return None
 
-# Function to load grammar evaluation pipelines.
+# Function to load grammar evaluation pipelines with appropriate device setting.
 def load_grammar_pipeline(model_name):
     try:
-        device = 0 if torch.cuda.is_available() else -1
         return pipeline("text2text-generation", model=model_name, device=device)
     except Exception as e:
         print(f"Error loading grammar model {model_name}: {e}")
@@ -76,12 +74,12 @@ def is_palindrome(text):
     cleaned = clean_text(text)
     return cleaned == cleaned[::-1]
 
-# Updated prompt: instruct output to contain only the palindrome.
+# Updated prompt instructs the model to output only the palindrome.
 def build_prompt(lang):
     return (
         f"Instruction: Generate a single original palindrome in {lang}.\n"
         "Output only the palindrome. The palindrome should be a continuous text that reads the same forward and backward.\n"
-        "Do not output any additional text, commentary, or the prompt itself.\n"
+        "Do not output any additional text or commentary.\n"
         "Palindrome: "
     )
 
@@ -144,6 +142,7 @@ def run_benchmark_all():
     print(f"CSV saved to {os.path.abspath(csv_path)}")
     return gr.Dataframe(df), csv_path
 
+# Build the Gradio UI using Blocks for a canvas layout.
 with gr.Blocks(title="Premium Model Palindrome Benchmark") as demo:
     gr.Markdown("# Premium Model Palindrome Benchmark")
     gr.Markdown(