Upload folder using huggingface_hub

chatbot.py

@@ -6,7 +6,7 @@ try:
     model_name = "gpt2"
 
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
+    model = AutoModelForCausalLM.from_pretrained(model_name)
 
     chat_history = [{"role": "system", "content": "You are a helpful assistant."}]