ethix commited on
Commit
2669b96
·
1 Parent(s): 0f71fbb

Refactor project interface and remove unused utility files

Browse files

- Update Gradio interface description to emphasize model evaluation
- Remove temporary slider and update project description
- Delete unused `goat.py` utility file and large model file
- Modify `.gitignore` to exclude additional files
- Comment out previously imported inference function

Files changed (4) hide show
  1. .gitignore +2 -0
  2. app.py +4 -11
  3. large_model_3lakh_v1.h5 +0 -3
  4. utils/goat.py +0 -15
.gitignore CHANGED
@@ -1 +1,3 @@
1
  .env
 
 
 
1
  .env
2
+ *goat.py
3
+ .vscode
app.py CHANGED
@@ -5,7 +5,7 @@ from torchvision import transforms
5
  import torch
6
  from PIL import Image
7
  import numpy as np
8
- from utils.goat import call_inference
9
  import io
10
  import warnings
11
 
@@ -503,7 +503,7 @@ def predict_image_with_html(img, confidence_threshold):
503
 
504
  # Define the Gradio interface
505
  with gr.Blocks() as iface:
506
- gr.Markdown("# AI Generated Image Classification")
507
 
508
  with gr.Row():
509
  with gr.Column(scale=1):
@@ -511,15 +511,8 @@ with gr.Blocks() as iface:
511
  confidence_slider = gr.Slider(0.0, 1.0, value=0.5, step=0.01, label="Confidence Threshold")
512
  inputs = [image_input, confidence_slider]
513
  with gr.Column(scale=2):
514
- with gr.Accordion("Project OpenSight", open=True):
515
- gr.Markdown("## Open source SOTA image moderation and detection tool.")
516
- temp_slider = gr.Slider(
517
- 0, 1,
518
- value=0.1,
519
- step=0.1,
520
- interactive=True,
521
- label="Slide me",
522
- )
523
  image_output = gr.Image(label="Processed Image", visible=False)
524
  # Custom HTML component to display results in 5 columns
525
  results_html = gr.HTML(label="Model Predictions")
 
5
  import torch
6
  from PIL import Image
7
  import numpy as np
8
+ # from utils.goat import call_inference / announcement soon
9
  import io
10
  import warnings
11
 
 
503
 
504
  # Define the Gradio interface
505
  with gr.Blocks() as iface:
506
+ gr.Markdown("# AI Generated Image / Deepfake Detection Models Evaluation")
507
 
508
  with gr.Row():
509
  with gr.Column(scale=1):
 
511
  confidence_slider = gr.Slider(0.0, 1.0, value=0.5, step=0.01, label="Confidence Threshold")
512
  inputs = [image_input, confidence_slider]
513
  with gr.Column(scale=2):
514
+ with gr.Accordion("Project OpenSight - Model Evaluations & Playground", open=True):
515
+ gr.Markdown("## OpenSight is a SOTA gen. image detection model, in pre-release prep.\n\nThis HF Space is a temporary home for us and the public to evaluate the shortcomings of current open source models.")
 
 
 
 
 
 
 
516
  image_output = gr.Image(label="Processed Image", visible=False)
517
  # Custom HTML component to display results in 5 columns
518
  results_html = gr.HTML(label="Model Predictions")
large_model_3lakh_v1.h5 DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c9fbb9bbfb644a42f077e20a2b4da3d7240879303474236e77d045b89fd46fff
3
- size 180867592
 
 
 
 
utils/goat.py DELETED
@@ -1,15 +0,0 @@
1
- import os
2
- import requests
3
- import json
4
-
5
- def call_inference(img):
6
- files = {
7
- 'image': ('image.jpg', img, 'image/jpeg')
8
- }
9
- print(img)
10
-
11
- response = requests.post(
12
- "https://aiwb-sightmodel.web.val.run",
13
- files=files
14
- )
15
- return response.json()