zmeeks commited on
Commit
f5e76a9
·
verified ·
1 Parent(s): 605b159

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ class BasicAgent:
65
  # Initialize the model
66
  print("🤖 Initializing TransformersModel...")
67
  self.model = TransformersModel(
68
- model_id="google/gemma-3-12b-it",
69
  torch_dtype=torch.bfloat16,
70
  device_map="auto",
71
  )
 
65
  # Initialize the model
66
  print("🤖 Initializing TransformersModel...")
67
  self.model = TransformersModel(
68
+ model_id="google/gemma-3-4b-it",
69
  torch_dtype=torch.bfloat16,
70
  device_map="auto",
71
  )