Spaces:
Sleeping
Sleeping
File size: 836 Bytes
1e722f3 19c6b02 ed170e8 13a3514 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import gradio as gr
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("sengeolab/geocode")
model = AutoModel.from_pretrained("sengeolab/geocode")
# Define your Gradio interface
inputs = gr.Textbox(lines=3, label="Input Address")
outputs = gr.Label(label="Geocoded Location")
def geocode_address(address):
# Tokenize the input text
inputs = tokenizer(address, return_tensors="pt", padding=True, truncation=True)
# Perform inference
outputs = model(**inputs)
# Add your inference logic here
# For example, you can process the outputs and return the geocoded location
return "Geocoded Location" # Replace this with your actual output
# Launch the Gradio interface
gr.Interface(fn=geocode_address, inputs=inputs, outputs=outputs).launch()
|