omarhkh commited on
Commit
02c19a7
·
1 Parent(s): 38164a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -86,7 +86,7 @@ def detect_objects2(model_name,url_input,image_input,threshold,type2):
86
  model = DetrForObjectDetection.from_pretrained(model_name)
87
 
88
 
89
-
90
 
91
  image = image_input
92
 
@@ -103,8 +103,7 @@ def detect_objects2(model_name,url_input,image_input,threshold,type2):
103
  if det_lab.count(1) > 0:
104
  total_text="Trench is Detected \n Image is Not Blurry \n"
105
  else:
106
- total_text="Trench is NOT Detected \n Image is Blurry \n"
107
- global xxresult
108
  xxresult=1
109
  print(type2)
110
  print(type(type2))
@@ -132,6 +131,7 @@ def tott():
132
  text2 = "The photo is ACCEPTED"
133
  else:
134
  text2 = "The photo is NOT ACCEPTED"
 
135
  return text2
136
 
137
  def set_example_image(example: list) -> dict:
@@ -141,14 +141,14 @@ def set_example_url(example: list) -> dict:
141
  return gr.Textbox.update(value=example[0])
142
 
143
 
144
- title = """<h1 id="title">Object Detection App with DETR and YOLOS</h1>"""
145
 
146
  description = """
147
- Links to HuggingFace Models:
148
- - [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50)
149
- - [facebook/detr-resnet-101](https://huggingface.co/facebook/detr-resnet-101)
150
- - [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small)
151
- - [hustvl/yolos-tiny](https://huggingface.co/hustvl/yolos-tiny)
152
  """
153
 
154
  models = ["omarhkh/detr-finetuned-omar8"]
 
86
  model = DetrForObjectDetection.from_pretrained(model_name)
87
 
88
 
89
+ global xxresult
90
 
91
  image = image_input
92
 
 
103
  if det_lab.count(1) > 0:
104
  total_text="Trench is Detected \n Image is Not Blurry \n"
105
  else:
106
+ total_text="Trench is NOT Detected \n Image is Blurry \n"
 
107
  xxresult=1
108
  print(type2)
109
  print(type(type2))
 
131
  text2 = "The photo is ACCEPTED"
132
  else:
133
  text2 = "The photo is NOT ACCEPTED"
134
+ xxresult==0
135
  return text2
136
 
137
  def set_example_image(example: list) -> dict:
 
141
  return gr.Textbox.update(value=example[0])
142
 
143
 
144
+ title = """<h1 id="title">Object Detection App for POC</h1>"""
145
 
146
  description = """
147
+ This application will test the model for the followings:
148
+ - Select the model
149
+ - Select the type of classification
150
+ - Then Detect
151
+ - The actual Results
152
  """
153
 
154
  models = ["omarhkh/detr-finetuned-omar8"]