Update space

app.py

@@ -4,6 +4,7 @@ model_name = "emeses/lab2_model"
 # Load the model with 8-bit quantization (works better on CPU)
 model = AutoModelForCausalLM.from_pretrained(model_name, quantization_config={"load_in_8bit": True})
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = model.to("cpu")
 
 # Function for Gradio interface
 def generate_text(prompt):