update requirements.txt

model/models/openai_api_models.py

@@ -25,6 +25,9 @@ def load_openai_model(model_name):
 
     image_url = response.data[0].url
 
+    return image_url
+
 
 if __name__ == "__main__":
-    load_openai_model('Dalle-2')
+    image_url = load_openai_model('Dalle-3')
+    print(image_url)

requirements.txt

@@ -8,7 +8,7 @@ h5py
 numpy>=1.23.5
 pandas<2.0.0
 peft
-torch==2.0.1
+torch==2.2.0
 torchvision==0.15.2 
 torchaudio==2.0.2
 jupyterlab>=4.0.2