Create app.py

app.py

@@ -0,0 +1,13 @@
+import torch
+from transformers import pipeline
+
+pipe = pipeline(
+    "text-generation",
+    model="google/gemma-2-2b",
+    device="cuda",  # replace with "mps" to run on a Mac device
+)
+
+text = "Once upon a time,"
+outputs = pipe(text, max_new_tokens=256)
+response = outputs[0]["generated_text"]
+print(response)