Update app.py

app.py

@@ -18,15 +18,15 @@ DESCRIPTION = """\
 This Space demonstrates model [DeepSeek-Coder](https://huggingface.co/deepseek-ai/deepseek-coder-6.7b-instruct) by DeepSeek, a code model with 6.7B parameters fine-tuned for chat instructions.
 """
 
-if not torch.cuda.is_available():
-    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
+# if not torch.cuda.is_available():
+#     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
 
-if torch.cuda.is_available():
-    model_id = "deepseek-ai/deepseek-coder-33b-instruct"
-    model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
-    tokenizer.use_default_system_prompt = False
+# if torch.cuda.is_available():
+#     model_id = "deepseek-ai/deepseek-coder-33b-instruct"
+#     model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
+#     tokenizer = AutoTokenizer.from_pretrained(model_id)
+#     tokenizer.use_default_system_prompt = False
     
 
 
@@ -121,7 +121,7 @@ chat_interface = gr.ChatInterface(
             value=1,
         ),
     ],
-    stop_btn=None,
+    stop_btn=True,
     examples=[
         ["implement snake game using pygame"],
         ["Can you explain briefly to me what is the Python programming language?"],