Update app.py

app.py

@@ -33,11 +33,10 @@ if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
 
-if torch.cuda.is_available():
-    model_id = "sunilswain/llama2-7b-chat-EssplTravelPolicy3.7k-epoch5"
-    model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
-    tokenizer.use_default_system_prompt = False
+model_id = "sunilswain/llama2-7b-chat-EssplTravelPolicy3.7k-epoch5"
+model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+tokenizer.use_default_system_prompt = False
 
 
 @spaces.GPU