test

app.py

@@ -1,3 +1,10 @@
-import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-gr.load("models/google/gemma-7b").launch()
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b")
+model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")
+
+input_text = "Write me a poem about Machine Learning."
+input_ids = tokenizer(input_text, return_tensors="pt")
+
+outputs = model.generate(**input_ids)
+print(tokenizer.decode(outputs[0]))