Create app.py

app.py

@@ -0,0 +1,30 @@
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("stabilityai/stable-code-3b")
+model = AutoModelForCausalLM.from_pretrained(
+  "stabilityai/stable-code-3b",
+  torch_dtype="auto",
+)
+
+
+codeSnip = """
+Is the following CSS code valid?
+```
+div {
+   color: red;
+   width: 100px;
+}
+```
+"""
+
+inputs = tokenizer(codeSnip, return_tensors="pt")
+
+tokens = model.generate(
+    **inputs,
+    max_new_tokens=48,
+    temperature=0.2,
+    do_sample=True,
+)
+
+res = tokenizer.decode(tokens[0], skip_special_tokens=True)
+print(res)