Trying with gpt2

app.py

@@ -21,9 +21,17 @@ from transformers import pipeline
 #OSError: We couldn't connect to 'https://huggingface.co' to load this file, couldn't find it in the cached files and it looks like WizardLM/WizardLM-7B-V1.0 is not the path to a directory containing a file named config.json.
 #Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/transformers/installation#offline-mode'.
 
-
 hub_model_id = "tiiuae/falcon-7b-instruct"
 response = pipeline("text-generation", model=hub_model_id)
+#obtuve este error
+#runtime error
+#Memory limit exceeded (16Gi)
+#obtuve este error - despues de hacer hw upgrade
+#runtime error
+#Memory limit exceeded (32Gi)
+
+response = pipeline('text-generation', model='gpt2')
+set_seed(42)
 
 def resp(text):
     return response(text)