Update app.py

app.py

@@ -1,11 +1,26 @@
+import torch
+import os
+import random
 import gradio as gr
+from transformers import pipeline
+import base64
+from datasets import load_dataset
+from diffusers import DiffusionPipeline
+from huggingface_hub import login
+import numpy as np
 
-def greet(name, intensity):
-    return "Hello, " + name + "!" * int(intensity)
+def guessanImage(image):
+    model = "microsoft/resnet-50"
+    st.write("using model:"+model)
+    imgclassifier  = pipeline("image-classification", model=model)
+    if image is not None:  
+        imgclassifier(image)
+        
+    return description
 
 demo = gr.Interface(
-    fn=greet,
-    inputs=["text", "slider"],
+    fn=guessanImage,
+    inputs=[gr.Image()],
     outputs=["text"],
 )