Update app.py

app.py

@@ -1,11 +1,38 @@
+"""
+# Inference
+
 import gradio as gr
 
-gr.load(
+app = gr.load(
     "google/gemma-2-2b-it",
     src = "models",
     inputs = [gr.Textbox(label = "Input")],
     outputs = [gr.Textbox(label = "Output")],
-    title = "Demo",
+    title = "Gemma",
+    examples = [
+        ["Hello, World."]
+    ]
+).launch()
+"""
+
+# Pipeline
+
+import gradio as gr
+from transformers import pipeline
+
+pipe = pipeline(model = "google/gemma-2-2b-it")
+
+def fn(input):
+    output = pipe(
+        input,
+    )
+    return output[0]["generated_text"]#[len(input):]
+
+app = gr.Interface(
+    fn = fn,
+    inputs = [gr.Textbox(label = "Input")],
+    outputs = [gr.Textbox(label = "Output")],
+    title = "Gemma",
     examples = [
         ["Hello, World."]
     ]