Update app.py

app.py

@@ -1,11 +1,15 @@
 from transformers import pipeline
 import gradio as gr
 
-pipeline = pipeline(model = "google/gemma-2b-it")
+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 = pipeline(input)
-    return output[0]['generated_text']
+    output = pipe1(
+        input
+    )
+    return output[0]['generated_text'][len(input):]
 
 app = gr.Interface(
     fn = inference,
@@ -15,6 +19,4 @@ app = gr.Interface(
     examples = [
         ["Hello, Gemma."]
     ]
-)
-
-app.launch()
+).launch()