update

app.py

@@ -9,7 +9,7 @@ if not huggingface_token:
     pass
     raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
 
-model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+model_id = "google/gemma-2-9b-it"
 #model_id = "microsoft/Phi-3-mini-128k-instruct"
 # device_map style value auto not cuda
 device = "auto" #torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -49,8 +49,8 @@ iface = gr.Interface(
         gr.Textbox(lines=2, label="System Message", value="You are a helpful assistant."),
     ],
     outputs=gr.Textbox(label="Generated Text"),
-    title="Llama 3.1 8B Instruct Text Generation",
-    description="Enter a prompt and optional system message to generate text using the Llama 3.1 8B Instruct model.",
+    title="google/gemma-2-9b-it Text Generation",
+    description="Enter a prompt and optional system message to generate text using the google/gemma-2-9b-it model.",
 )
 
 if __name__ == "__main__":