sengeolab commited on
Commit
ed170e8
·
verified ·
1 Parent(s): 673a691

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -1,14 +1,16 @@
1
 
2
  import gradio as gr
3
 
4
- #gr.load("models/sengeolab/geocoder").launch()
5
- #import gradio as gr
6
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
7
-
8
- # Load the model and tokenizer
9
- model_name = "sengeolab/geocode" # This should be the correct model identifier
10
- tokenizer = AutoTokenizer.from_pretrained(model_name)
11
- model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
 
12
 
13
  # Define your Gradio interface
14
  inputs = gr.Textbox(lines=3, label="Input Address")
 
1
 
2
  import gradio as gr
3
 
4
+ from transformers import pipeline
5
+
6
+ pipe = pipeline("sentence-similarity", model="sengeolab/geocode")
7
+
8
+ Copy
9
+ # Load model directly
10
+ from transformers import AutoTokenizer, AutoModel
11
+
12
+ tokenizer = AutoTokenizer.from_pretrained("sengeolab/geocode")
13
+ model = AutoModel.from_pretrained("sengeolab/geocode")
14
 
15
  # Define your Gradio interface
16
  inputs = gr.Textbox(lines=3, label="Input Address")