Mori-kamiyama commited on
Commit
778a3e4
·
1 Parent(s): fc3790e

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # GPU で動作するモデルのロード(自動でGPUがあれば使用)
5
- generator = pipeline("text-generation", model="Mori-kamiyama/sarashina2-13b-r1")
6
 
7
  def generate_text(prompt):
8
  result = generator(prompt, max_length=100, num_return_sequences=1)
@@ -11,6 +11,6 @@ def generate_text(prompt):
11
  iface = gr.Interface(fn=generate_text,
12
  inputs="text",
13
  outputs="text",
14
- title="Text Generator using GPT-2")
15
 
16
  iface.launch()
 
2
  from transformers import pipeline
3
 
4
  # GPU で動作するモデルのロード(自動でGPUがあれば使用)
5
+ generator = pipeline("text-generation", model="google/gemma-3-4b-it")
6
 
7
  def generate_text(prompt):
8
  result = generator(prompt, max_length=100, num_return_sequences=1)
 
11
  iface = gr.Interface(fn=generate_text,
12
  inputs="text",
13
  outputs="text",
14
+ title="Text Generator using sarashina2-13b-r1")
15
 
16
  iface.launch()