Spaces:
Runtime error
Runtime error
File size: 1,049 Bytes
5b55bc2 fd27a87 f9466b5 fd27a87 6e8f934 ff58162 fd27a87 6e8f934 fd27a87 6e8f934 |
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 28 29 30 |
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras.preprocessing import image_dataset_from_directory
from tensorflow.keras import models, layers
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import gradio as gr
from transformers import AutoTokenizer, AutoModelForImageClassification
from PIL import Image
import requests
import torch
# Load model from Hugging Face model hub
model_name = "NKASG/GNN" # Replace with your model's name on Hugging Face
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForImageClassification.from_pretrained(model_name)
# Define function for image preprocessing and prediction
def process_image(image):
img_4d=img.reshape(-1,256,256,3)
prediction=model.predict(img_4d)[0]
# return {class_names[i]: float(prediction[i]) for i in range(3)}
image = gr.inputs.Image(shape=(256,256))
label = gr.outputs.Label(num_top_classes=1)
gr.Interface(fn=predict_image, inputs=image, outputs=label,interpretation='default').launch()
|