Update Dockerfile

app.py

@@ -0,0 +1,13 @@
+
+import gradio as gr
+from huggingface_hub import InferenceClient
+
+client = InferenceClient("google/gemma-2b-it")
+
+def generate_text(prompt):
+    response = client.text_generation(prompt, max_new_tokens=50)
+    return response
+
+demo = gr.Interface(fn=generate_text, inputs="text", outputs="text")
+demo.launch()
+