Update app.py

app.py

@@ -9,7 +9,9 @@ from synthid_text import detector_bayesian, synthid_mixin, logits_processing
 MODEL_NAME = "google/gemma-7b-it"  # Choose the model (Gemma models used in SynthID)
 DEVICE = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
 
-# Initialize tokenizer and model configuration
+
+# Initialize model and tokenizer
+model = transformers.AutoModelForCausalLM.from_pretrained(MODEL_NAME).to(DEVICE)
 tokenizer = transformers.AutoTokenizer.from_pretrained(MODEL_NAME)
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = "left"