Update app.py

app.py

@@ -17,7 +17,7 @@ if HUGGINGFACE:
     MODEL_NAME="xu3kev/deepseekcoder-7b-logo-pbe"
     import torch
     from transformers import AutoModelForCausalLM, AutoTokenizer
-    hug_model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16, attn_implementation="flash_attention_2",).to('cuda')
+    hug_model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16,).to('cuda')
     hug_tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
 INPUT_STRUCTION_TEMPLATE = """Here is a gray scale images representing with integer values 0-9.