Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
-from datasets import load_dataset
+#from datasets import load_dataset
 from transformers import pipeline
 
-dataset = load_dataset("rebase/hello")
+#dataset = load_dataset("rebase/hello")
 model = pipeline("text-generation")
 
 def image_classifier(inp):
-    return dataset
+    return "dataset"
 
 def predict(prompt):
     completion = model(prompt)[0]["generated_text"]