jpagotto's picture
Update app.py
4cc0596
raw
history blame contribute delete
No virus
983 Bytes
import gradio as gr
from fastai.vision.all import *
from fastcore.all import *
import skimage
import io
import pandas as pd
df = pd.read_csv('citytempdata.csv')
def tempfromPath(input_Path):
a = parent_label(input_Path)
#print(a)
b = float(df[df['Capital']==a]['Temperature (degC)'])
return b
learn = load_learner('P1_model_v2.pkl')
def predict(img):
img = PILImage.create(img)
Temp = learn.predict(img)
return f"""Mean temperature is {round(Temp[0][0],1)} degrees celsius."""
title = "City Temperature Estimater"
description = "A fastai model trained on aerial views of cities in order to estimate their mean annual temperatures."
examples = ['Brisbane_aerial_view.jpg','NewYork_aerial_view.jpg','Yakutsk_aerial_view.jpg','LasVegas_aerial_view.jpg','Darwin_aerial_view.jpg']
gr.Interface(fn=predict,inputs=gr.inputs.Image(shape=(512,512)),outputs=gr.outputs.Textbox(),title=title,description=description,examples=examples).launch()