more info displayed

app.py

@@ -12,6 +12,12 @@ def load_class_names(file_path):
     with open(file_path, 'r') as f:
         class_names = [line.strip() for line in f.readlines()]
     return class_names
+# Function to load Wikipedia links from a file
+def load_class_info(file_path):
+    with open(file_path, 'r') as f:
+        class_info = [line.strip() for line in f.readlines()]
+    return class_info
+
 # Function to load the model from a .pkl file
 def load_model(model_path, model_type='resnet'):
     # Load the model state dictionary
@@ -58,13 +64,18 @@ def predict_image(image):
         confidence, preds = torch.max(probabilities, 0)
     
     class_name = class_names[preds.item()]
+    wiki_link = class_info[preds.item()]
     confidence_score = confidence.item() * 100
-    return f"{class_name}: {confidence_score:.2f}%"
+    result = f"{class_name}: {confidence_score:.2f}%"
+    html_result = f"<h1><br><a href='{wiki_link}' target='_blank'>More Info</a></h1>"
+    return result, html_result
 
 # Load trained model and class names
 model_path = 'resnet30EpochsPretrainedNFeatureX_model.pkl'
 class_file_path = 'classes.txt'
+class_info_path = 'classinfo.txt'
 class_names = load_class_names(class_file_path)
+class_info = load_class_info(class_info_path)
 num_classes = len(class_names)
 model = load_model(model_path)
 
@@ -72,7 +83,7 @@ model = load_model(model_path)
 iface = gr.Interface(
     fn=predict_image, 
     inputs=gr.Image(),
-    outputs=gr.Label(num_top_classes=1),
+    outputs=[gr.Label(num_top_classes=1),gr.HTML()],
     title="Image Classification",
     description="Upload an image to get the predicted label",
     allow_flagging="never",

classinfo.txt

@@ -0,0 +1,90 @@
+https://en.wikipedia.org/wiki/antelope
+https://en.wikipedia.org/wiki/badger
+https://en.wikipedia.org/wiki/bat
+https://en.wikipedia.org/wiki/bear
+https://en.wikipedia.org/wiki/bee
+https://en.wikipedia.org/wiki/beetle
+https://en.wikipedia.org/wiki/bison
+https://en.wikipedia.org/wiki/boar
+https://en.wikipedia.org/wiki/butterfly
+https://en.wikipedia.org/wiki/cat
+https://en.wikipedia.org/wiki/caterpillar
+https://en.wikipedia.org/wiki/chimpanzee
+https://en.wikipedia.org/wiki/cockroach
+https://en.wikipedia.org/wiki/cow
+https://en.wikipedia.org/wiki/coyote
+https://en.wikipedia.org/wiki/crab
+https://en.wikipedia.org/wiki/crow
+https://en.wikipedia.org/wiki/deer
+https://en.wikipedia.org/wiki/dog
+https://en.wikipedia.org/wiki/dolphin
+https://en.wikipedia.org/wiki/donkey
+https://en.wikipedia.org/wiki/dragonfly
+https://en.wikipedia.org/wiki/duck
+https://en.wikipedia.org/wiki/eagle
+https://en.wikipedia.org/wiki/elephant
+https://en.wikipedia.org/wiki/flamingo
+https://en.wikipedia.org/wiki/fly
+https://en.wikipedia.org/wiki/fox
+https://en.wikipedia.org/wiki/goat
+https://en.wikipedia.org/wiki/goldfish
+https://en.wikipedia.org/wiki/goose
+https://en.wikipedia.org/wiki/gorilla
+https://en.wikipedia.org/wiki/grasshopper
+https://en.wikipedia.org/wiki/hamster
+https://en.wikipedia.org/wiki/hare
+https://en.wikipedia.org/wiki/hedgehog
+https://en.wikipedia.org/wiki/hippopotamus
+https://en.wikipedia.org/wiki/hornbill
+https://en.wikipedia.org/wiki/horse
+https://en.wikipedia.org/wiki/hummingbird
+https://en.wikipedia.org/wiki/hyena
+https://en.wikipedia.org/wiki/jellyfish
+https://en.wikipedia.org/wiki/kangaroo
+https://en.wikipedia.org/wiki/koala
+https://en.wikipedia.org/wiki/ladybugs
+https://en.wikipedia.org/wiki/leopard
+https://en.wikipedia.org/wiki/lion
+https://en.wikipedia.org/wiki/lizard
+https://en.wikipedia.org/wiki/lobster
+https://en.wikipedia.org/wiki/mosquito
+https://en.wikipedia.org/wiki/moth
+https://en.wikipedia.org/wiki/mouse
+https://en.wikipedia.org/wiki/octopus
+https://en.wikipedia.org/wiki/okapi
+https://en.wikipedia.org/wiki/orangutan
+https://en.wikipedia.org/wiki/otter
+https://en.wikipedia.org/wiki/owl
+https://en.wikipedia.org/wiki/ox
+https://en.wikipedia.org/wiki/oyster
+https://en.wikipedia.org/wiki/panda
+https://en.wikipedia.org/wiki/parrot
+https://en.wikipedia.org/wiki/pelecaniformes
+https://en.wikipedia.org/wiki/penguin
+https://en.wikipedia.org/wiki/pig
+https://en.wikipedia.org/wiki/pigeon
+https://en.wikipedia.org/wiki/porcupine
+https://en.wikipedia.org/wiki/possum
+https://en.wikipedia.org/wiki/raccoon
+https://en.wikipedia.org/wiki/rat
+https://en.wikipedia.org/wiki/reindeer
+https://en.wikipedia.org/wiki/rhinoceros
+https://en.wikipedia.org/wiki/sandpiper
+https://en.wikipedia.org/wiki/seahorse
+https://en.wikipedia.org/wiki/Pinniped
+https://en.wikipedia.org/wiki/shark
+https://en.wikipedia.org/wiki/sheep
+https://en.wikipedia.org/wiki/snake
+https://en.wikipedia.org/wiki/sparrow
+https://en.wikipedia.org/wiki/squid
+https://en.wikipedia.org/wiki/squirrel
+https://en.wikipedia.org/wiki/starfish
+https://en.wikipedia.org/wiki/swan
+https://en.wikipedia.org/wiki/tiger
+https://en.wikipedia.org/wiki/turkey
+https://en.wikipedia.org/wiki/turtle
+https://en.wikipedia.org/wiki/whale
+https://en.wikipedia.org/wiki/wolf
+https://en.wikipedia.org/wiki/wombat
+https://en.wikipedia.org/wiki/woodpecker
+https://en.wikipedia.org/wiki/zebra