Update app.py

app.py

@@ -7,17 +7,17 @@ def fmt_prompt(prompt: str) -> str:
     return f"""[Instructions]:\n{prompt}\n\n[Response]:"""
 
 
-    
-    model_name = "abacaj/starcoderbase-1b-sft"
-    tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
 
-    model = (
-        transformers.AutoModelForCausalLM.from_pretrained(
-            model_name,
-        )
-        .to("cuda:0")
-        .eval()
+model_name = "abacaj/starcoderbase-1b-sft"
+tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
+
+model = (
+    transformers.AutoModelForCausalLM.from_pretrained(
+        model_name,
     )
+    .to("cuda:0")
+    .eval()
+)
 def chat_fn(prompt):
     #prompt = "Write a python function to sort the following array in ascending order, don't use any built in sorting methods: [9,2,8,1,5]"
     prompt_input = fmt_prompt(prompt)