Update app.py

app.py

@@ -37,17 +37,17 @@ def classify_website(site_text):
         
         prompt = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
     
-    ### Instruction:
-    Categorize the website into one of the 3 categories:
-    
-    1) OTHER
-    2) NEWS/BLOG
-    3) E-commerce
-    
-    ### Input:
-    {site_text}
-    
-    ### Response:"""
+### Instruction:
+Categorize the website into one of the 3 categories:
+
+1) OTHER
+2) NEWS/BLOG
+3) E-commerce
+
+### Input:
+{site_text}
+
+### Response:"""
         
         inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
         outputs = model.generate(**inputs, max_new_tokens=64, use_cache=True)