Refactor app.py to remove device_map parameter from model initialization, simplifying the configuration for text generation. This change enhances code clarity and focuses on essential parameters.

app.py

@@ -65,7 +65,6 @@ if access_token:
             model = AutoModelForCausalLM.from_pretrained(
                 "google/gemma-2b-it",
                 token=access_token,
-                device_map="auto"  # Better device handling
             )
             return pipeline(
                 "text-generation",