Update myapp.py

myapp.py

@@ -17,7 +17,7 @@ model_name = "prompthero/openjourney-v4"  # Replace with your model name
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Load the model weights from safeload
-model = AutoModel.from_pretrained(model_name, state_dict=model_data)
+model = AutoModel.from_pretrained(model_name, state_dict=model_data).to("cpu")
 
 @myapp.route('/')
 def index():