Update app.py

app.py

@@ -1,18 +1,8 @@
-from transformers import pipeline
 import gradio as gr
 
-pipe1 = pipeline(model = "google/gemma-2b-it")
-#pipe2 = pipeline(model = "google/gemma-2-9b-it")
-#pipe3 = pipeline(model = "google/gemma-2-27b-it")
-
-def inference(input):
-    output = pipe1(
-        input
-    )
-    return output[0]['generated_text'][len(input):]
-
-app = gr.Interface(
-    fn = inference,
+app = gr.load(
+    google/gemma-2-2b-it,
+    src = "models",
     inputs = [gr.Textbox(label = "Input")],
     outputs = [gr.Textbox(label = "Output")],
     title = "Demo",