Update app.py

app.py

@@ -24,7 +24,7 @@ from transformers import AutoTokenizer, AutoModel
 hf_token = os.environ.get("llmtoken")  # HF_TOKEN is the name of the secret in your Space
 
 # Step 2: Specify the model
-model_name =  "Qwen/Qwen3-4B-Instruct-2507"
+model_name =  "gpt2"
 
 
 # Step 3: Load tokenizer and model using the token
@@ -43,13 +43,13 @@ arg_dict = {
     'all_models':[
         # "meta-llama/Llama-3.1-8B", # too big for the A10G 24GB
         #"meta-llama/Llama-3.2-3B",
-       # "gpt2",
+        "gpt2",
         # "meta-llama/Llama-3.2-1B",
         # "Qwen/Qwen3-8B", # too big for the A10G 24GB
         # "Qwen/Qwen3-4B",
         # "Qwen/Qwen3-1.7B",
         # "Qwen/Qwen3-0.6B",
-      "Qwen/Qwen3-4B-Instruct-2507",
+      #"Qwen/Qwen3-4B-Instruct-2507",
         # "Qwen/Qwen3-4B-Thinking-2507",
     ],
     # 'load_fp16' : True,