feat: add basic files

.gitignore

@@ -0,0 +1,2 @@
+venv
+flagged

auto_train.py

@@ -0,0 +1,29 @@
+import gradio as gr
+from transformers import pipeline
+
+raw_model_name = 'distilbert-base-uncased'
+raw_model = pipeline('sentiment-analysis', model=raw_model_name)
+fine_tuned_model_name = 'distilbert-base-uncased-finetuned-sst-2-english'
+fine_tuned_model = pipeline('sentiment-analysis', model=fine_tuned_model_name)
+
+def get_model_output(input_text, model_choice):
+    raw_result = raw_model(input_text)
+    fine_tuned_result = fine_tuned_model(input_text)
+    return format_model_output(raw_result[0]), format_model_output(fine_tuned_result[0])
+
+def format_model_output(output):
+    return f"I am {output['score']*100:.2f}% sure that the sentiment is {output['label']}"
+
+iface = gr.Interface(
+    fn=get_model_output,
+    title="DistilBERT Sentiment Analysis",
+    inputs=[
+        gr.Textbox(label="Input Text"),
+    ],
+    outputs=[
+        gr.Textbox(label="Base DistilBERT output (distilbert-base-uncased)"),
+        gr.Textbox(label="Fine-tuned DistilBERT output")
+    ],
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,74 @@
+aiofiles==23.2.1
+altair==5.1.2
+annotated-types==0.6.0
+anyio==3.7.1
+attrs==23.1.0
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.3
+colorama==0.4.6
+contourpy==1.2.0
+cycler==0.12.1
+exceptiongroup==1.2.0
+fastapi==0.104.1
+ffmpy==0.3.1
+filelock==3.13.1
+fonttools==4.45.0
+fsspec==2023.10.0
+gradio==4.5.0
+gradio_client==0.7.0
+h11==0.14.0
+httpcore==1.0.2
+httpx==0.25.1
+huggingface-hub==0.19.4
+idna==3.4
+importlib-resources==6.1.1
+Jinja2==3.1.2
+jsonschema==4.20.0
+jsonschema-specifications==2023.11.1
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.2.1
+numpy==1.26.2
+orjson==3.9.10
+packaging==23.2
+pandas==2.1.3
+Pillow==10.1.0
+pydantic==2.5.1
+pydantic_core==2.14.3
+pydub==0.25.1
+Pygments==2.17.1
+pyparsing==3.1.1
+pystache==0.6.0
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+referencing==0.31.0
+regex==2023.10.3
+requests==2.31.0
+rich==13.7.0
+rpds-py==0.13.1
+safetensors==0.4.0
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+sympy==1.12
+tokenizers==0.15.0
+tomlkit==0.12.0
+toolz==0.12.0
+torch==2.1.1
+tqdm==4.66.1
+transformers==4.35.2
+typer==0.9.0
+typing_extensions==4.8.0
+tzdata==2023.3
+urllib3==2.1.0
+uvicorn==0.24.0.post1
+websockets==11.0.3