Update app.py

app.py

@@ -10,10 +10,10 @@ from io import BytesIO
 import base64
 import traceback
 
-# Function to run the Llama 4 text test
+# Function to run the Llama 4 Scout text test
 def test_text_generation():
     results = []
-    results.append("=== Testing Text Generation ===")
+    results.append("=== Testing Llama 4 Scout Text Generation ===")
     
     try:
         # Get token from environment
@@ -32,8 +32,8 @@ def test_text_generation():
             results.append(f"Error logging in: {e}")
             return "\n".join(results)
         
-        # Use a Llama 4 model that you have access to
-        model_id = "meta-llama/Llama-4-Maverick-17B-128E-Instruct"  # This is in your list
+        # Use Llama 4 Scout model
+        model_id = "meta-llama/Llama-4-Scout-17B-16E-Instruct"
         
         results.append(f"Loading tokenizer from {model_id}...")
         from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -76,7 +76,7 @@ def test_text_generation():
 # Function to run the Llama 4 Scout image-text test
 def test_image_text_generation():
     results = []
-    results.append("=== Testing Image-Text Generation ===")
+    results.append("=== Testing Llama 4 Scout Image-Text Generation ===")
     
     try:
         # Get token from environment
@@ -95,8 +95,8 @@ def test_image_text_generation():
             results.append(f"Error logging in: {e}")
             return "\n".join(results)
         
-        # Use the Llama 4 Scout model that you have access to
-        model_id = "meta-llama/Llama-4-Scout-17B-16E-Instruct"  # This is in your list
+        # Use the Llama 4 Scout model
+        model_id = "meta-llama/Llama-4-Scout-17B-16E-Instruct"
         
         results.append(f"Loading processor and model from {model_id}...")
         
@@ -187,10 +187,10 @@ def test_image_text_generation():
     
     return "\n".join(results)
 
-# Function to list available Llama models
-def list_llama_models():
+# Function to list available Llama Scout models
+def list_scout_models():
     results = []
-    results.append("=== Listing Available Llama Models ===")
+    results.append("=== Listing Available Llama 4 Scout Models ===")
     
     try:
         # Get token from environment
@@ -214,14 +214,15 @@ def list_llama_models():
         api = HfApi(token=token)
         
         results.append("Fetching models from meta-llama organization...")
-        results.append("Available models:")
+        results.append("Available Scout models:")
         
         model_count = 0
         for model in api.list_models(author="meta-llama"):
-            results.append(f"- {model.id}")
-            model_count += 1
+            if "Scout" in model.id:  # Only show Scout models
+                results.append(f"- {model.id}")
+                model_count += 1
         
-        results.append(f"Total models found: {model_count}")
+        results.append(f"Total Scout models found: {model_count}")
         results.append("MODEL LISTING SUCCESSFUL!")
         
     except Exception as e:
@@ -234,17 +235,17 @@ def list_llama_models():
 # Create Gradio interface
 with gr.Blocks(title="Llama 4 Scout Test") as demo:
     gr.Markdown("# Llama 4 Scout Test")
-    gr.Markdown("This Space tests the connection to Llama 4 and Llama 4 Scout models.")
+    gr.Markdown("This Space tests the connection to Llama 4 Scout models.")
     
-    with gr.Tab("List Available Models"):
+    with gr.Tab("List Scout Models"):
         with gr.Row():
             with gr.Column():
-                list_models_button = gr.Button("List Available Llama Models")
+                list_models_button = gr.Button("List Llama 4 Scout Models")
             with gr.Column():
-                models_result = gr.Textbox(label="Available Models", lines=20)
+                models_result = gr.Textbox(label="Available Scout Models", lines=20)
         
         list_models_button.click(
-            fn=list_llama_models,
+            fn=list_scout_models,
             inputs=[],
             outputs=[models_result]
         )
@@ -279,10 +280,10 @@ with gr.Blocks(title="Llama 4 Scout Test") as demo:
         gr.Markdown("""
         ## About This Test
         
-        This test checks if your Space can connect to and use Llama 4 and Llama 4 Scout models.
+        This test checks if your Space can connect to and use Llama 4 Scout models.
         
-        - The **List Available Models** tab shows all models available from meta-llama
-        - The **Text Generation Test** uses Llama 4 Maverick for basic text generation
+        - The **List Scout Models** tab shows all Scout models available from meta-llama
+        - The **Text Generation Test** uses Llama 4 Scout for basic text generation
         - The **Image-Text Generation Test** uses Llama 4 Scout for image-text generation
         
         If both tests pass, your Llama 4 Scout setup should work correctly.