Update main.py

main.py

@@ -15,7 +15,7 @@ if not gemini_api_key:
 # Make sure to use a model you have access to
 generator: TextGenerationPipeline = pipeline(
     "text-generation",
-    model="gemini-1.5-flash",  # Replace if needed
+    model="google/gemma-2-2b-it",  # Replace if needed
 )  # IMPORTANT: **DO NOT** set `use_auth_token` here
 
 # Data model for the request body