junaidbaber commited on
Commit
4257e69
·
verified ·
1 Parent(s): 97a2367

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ def initialize_model():
7
  """Initialize a small and fast model for CPU"""
8
  # Using a tiny model optimized for CPU
9
  model_id = "facebook/opt-125m" # Much smaller model (125M parameters)
 
10
 
11
  try:
12
  # Initialize the pipeline directly - more efficient than loading model separately
 
7
  """Initialize a small and fast model for CPU"""
8
  # Using a tiny model optimized for CPU
9
  model_id = "facebook/opt-125m" # Much smaller model (125M parameters)
10
+ model_id ="GEB-AGI/geb-1.3b"
11
 
12
  try:
13
  # Initialize the pipeline directly - more efficient than loading model separately