deeksonparlma commited on
Commit
5ab824d
·
1 Parent(s): 8ce028d
Files changed (3) hide show
  1. app.py +9 -5
  2. model.pkl +2 -2
  3. vectorizer.pkl +3 -0
app.py CHANGED
@@ -1,10 +1,14 @@
1
  import gradio as gr
2
  import pickle
3
 
4
- loaded_model = pickle.load(open("model.pkl", "rb"))
5
- def greet(img):
6
- pred = loaded_model(img)
7
- return pred
8
 
9
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 
 
 
 
 
10
  iface.launch()
 
1
  import gradio as gr
2
  import pickle
3
 
4
+ model = pickle.load(open("model.pkl", "rb"))
5
+ vectorizer = pickle.load(open("vectorizer.pkl", "rb"))
 
 
6
 
7
+ def greet(description):
8
+ new_query = description
9
+ new_issue_vector = vectorizer.transform([new_query])
10
+ prediction = model.predict(new_issue_vector)
11
+ return prediction
12
+
13
+ iface = gr.Interface(fn=greet, inputs="text", outputs="label",title="Farming Tracker")
14
  iface.launch()
model.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b18274debd00d65007d523f2c93b05ab6e4d7727dc904ead01758991bbe1dead
3
- size 42
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c49bbcbb28a4493047f0a3301c1dccdb177c2ed183d8e964212cdb6a5e8fb1e
3
+ size 45768
vectorizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e338f5b4a1162ea5b1f628cfe6d7cd727e4dc0a6c185d3110415561f6ae6e1e1
3
+ size 4512