Update app.py

app.py

@@ -6,11 +6,11 @@ from threading import Thread
 import spaces
 
 token = os.environ["HF_TOKEN"]
-model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it", 
+model = AutoModelForCausalLM.from_pretrained("google/gemma-1.1-2b-it", 
                                              # torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
                                              torch_dtype=torch.float16,
                                              token=token)
-tok = AutoTokenizer.from_pretrained("google/gemma-2b-it",token=token)
+tok = AutoTokenizer.from_pretrained("google/gemma-1.1-2b-it",token=token)
 # using CUDA for an optimal experience
 # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 if torch.cuda.is_available():