.

app.py

@@ -13,12 +13,11 @@ HUGGINGFACE=True
 MOCK = False
 TEST_FOLDER = "c4f5"
 
-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,).to('cuda')
-    hug_tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+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,).to('cuda')
+hug_tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
 INPUT_STRUCTION_TEMPLATE = """Here is a gray scale images representing with integer values 0-9.
 {image_str}