appvoid commited on
Commit
97831d4
Β·
verified Β·
1 Parent(s): 166b80d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import torch
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
  from transformers import pipeline
9
 
10
- model = pipeline("text-generation" , model="appvoid/arco-2")
11
 
12
  @spaces.GPU
13
  def predict(prompt):
@@ -18,5 +18,5 @@ gr.Interface(
18
  fn=predict,
19
  inputs="text",
20
  outputs="text",
21
- title="arco 2",
22
  ).launch()
 
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
  from transformers import pipeline
9
 
10
+ model = pipeline("text-generation" , model="appvoid/text-arco")
11
 
12
  @spaces.GPU
13
  def predict(prompt):
 
18
  fn=predict,
19
  inputs="text",
20
  outputs="text",
21
+ title="text arco",
22
  ).launch()