AI Space Builder: Space content update for prometheus04/my-ai-space

.gitattributes

@@ -1,35 +1 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+* text=auto eol=lf

README.md

@@ -1,12 +1,11 @@
 ---
 title: My Ai Space
-emoji: 👀
-colorFrom: purple
-colorTo: red
+emoji: 😻
+colorFrom: indigo
+colorTo: pink
 sdk: gradio
-sdk_version: 5.37.0
-app_file: app.py
+sdk_version: 5.32.1
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,7 @@
+import gradio as gr
+import torch
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+from app.models.predict import predict
+
+app = gr.Interface(fn=predict, inputs="text", outputs="label")
+app.launch()

app/models/config.json

@@ -0,0 +1,3 @@
+{
+  "model_name": "distilbert-base-uncased-finetuned-sst-2-english"
+}

app/models/config.py

@@ -0,0 +1,17 @@
+import json
+import Path
+from pathlib import Path
+
+class ModelConfig:
+    def __init__(self, json_file_path: str):
+        self.json_file_path = json_file_path
+
+    @classmethod
+    def from_json_file(cls, json_file_path: str):
+        with open(json_file_path, "r") as file:
+            config_data = json.load(file)
+        return cls(json_file_path)
+
+    def get_hyperparameters(self):
+        # implement your logic to retrieve hyperparameters from the JSON file
+        pass

app/models/predict.py

@@ -0,0 +1,15 @@
+import torch
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import json
+
+model_name = "distilbert-base-uncased-finetuned-sst-2-english"
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+def predict(text):
+    inputs = tokenizer(text, return_tensors="pt")
+    outputs = model(inputs["input_ids"], attention_mask=inputs["attention_mask"])
+    logits = outputs.logits
+    probabilities = torch.softmax(logits, dim=-1)
+    predicted_class = torch.argmax(probabilities)
+    return {"text": text, "prediction": predicted_class.item(), "probability": probabilities.tolist()}

app/requirements.txt

@@ -0,0 +1,4 @@
+transformers
+torch
+gradio
+tqdm

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch
+gradio