Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 def get_model(model_url="thepowerfuldeez/Qwen2-1.5B-Summarize", use_cpu=False):
     tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-1.5B-Instruct")
     if use_cpu:
-        model = OVModelForCausalLM.from_pretrained(
+        model = AutoModelForCausalLM.from_pretrained(
             model_url,
             device_map="cpu",
             load_in_4bit=True,