Update app.py

app.py

@@ -9,7 +9,7 @@ class LlamaGuardModeration:
         self.model = None
         self.tokenizer = None
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.model_id = "meta-llama/Llama-Guard-3-8B"
+        self.model_id = "meta-llama/Llama-Guard-3-8B"  # モデルIDを変更
         self.dtype = torch.bfloat16
         
         # HuggingFace tokenの取得
@@ -29,7 +29,7 @@ class LlamaGuardModeration:
                 token=self.huggingface_token
             )
             
-            # モデルの初期化（bitsandbytesなし）
+            # モデルの初期化
             self.model = AutoModelForCausalLM.from_pretrained(
                 self.model_id,
                 torch_dtype=self.dtype,