Pedro Bento commited on
Commit
beb80fa
·
1 Parent(s): 5278818

Added Virus total interface

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. tdagent/tools/virus_total.py +1 -1
app.py CHANGED
@@ -3,12 +3,14 @@ import gradio as gr
3
  from tdagent.tools.get_url_content import gr_get_url_http_content
4
  from tdagent.tools.letter_counter import gr_letter_counter
5
  from tdagent.tools.query_abuse_ip_db import gr_query_abuseipdb
 
6
 
7
  gr_app = gr.TabbedInterface(
8
  [
9
  gr_get_url_http_content,
10
  gr_letter_counter,
11
  gr_query_abuseipdb,
 
12
  ],
13
  )
14
 
 
3
  from tdagent.tools.get_url_content import gr_get_url_http_content
4
  from tdagent.tools.letter_counter import gr_letter_counter
5
  from tdagent.tools.query_abuse_ip_db import gr_query_abuseipdb
6
+ from tdagent.tools.virus_total import gr_virus_total
7
 
8
  gr_app = gr.TabbedInterface(
9
  [
10
  gr_get_url_http_content,
11
  gr_letter_counter,
12
  gr_query_abuseipdb,
13
+ gr_virus_total,
14
  ],
15
  )
16
 
tdagent/tools/virus_total.py CHANGED
@@ -45,7 +45,7 @@ Times Submitted: {url_analysis.times_submitted}
45
  return f"Error: {str(e)}"
46
 
47
 
48
- gr_query_abuseipdb = gr.Interface(
49
  fn=get_url_info,
50
  inputs=["text"],
51
  outputs="text",
 
45
  return f"Error: {str(e)}"
46
 
47
 
48
+ gr_virus_total = gr.Interface(
49
  fn=get_url_info,
50
  inputs=["text"],
51
  outputs="text",