Update app.py

app.py

@@ -6,14 +6,14 @@ import torch
 model_name = "bigcode/starcoder2-15b-instruct-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
-    model_name, 
-    torch_dtype=torch.float16,
-    device_map="auto"
+    model_name,
+    torch_dtype=torch.float16  # Vous pouvez aussi utiliser torch.float32 pour le CPU
 )
 
 # Fonction pour générer du texte
 def generate_text(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+    # Utiliser le CPU au lieu du GPU
+    inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(inputs["input_ids"], max_length=200)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)