Create app.py

app.py

@@ -0,0 +1,28 @@
+from sagemaker.huggingface import HuggingFaceModel
+import sagemaker
+
+role = sagemaker.get_execution_role()
+# Hub Model configuration. https://huggingface.co/models
+hub = {
+	'HF_MODEL_ID':'dalle-mini/dalle-mega',
+	'HF_TASK':'text2text-generation'
+}
+
+# create Hugging Face Model Class
+huggingface_model = HuggingFaceModel(
+	transformers_version='4.17.0',
+	pytorch_version='1.10.2',
+	py_version='py38',
+	env=hub,
+	role=role, 
+)
+
+# deploy model to SageMaker Inference
+predictor = huggingface_model.deploy(
+	initial_instance_count=1, # number of instances
+	instance_type='ml.m5.xlarge' # ec2 instance type
+)
+
+predictor.predict({
+	'inputs': No input example has been defined for this model task.
+})