MudassirFayaz commited on
Commit
0b30140
1 Parent(s): 83dc18a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -10,7 +10,10 @@ def greet_json():
10
  # pipe = pipeline("text2text-generation", model="google/flan-t5-small")
11
  # checking for ofingpt
12
  # ofintech/FinGPT_0.1.3
13
- pipe = pipeline("text2text-generation", model="MudassirFayaz/llama-2-7b_career_0.6.0", trust_remote_code=True)
 
 
 
14
 
15
 
16
  @app.get("/")
 
10
  # pipe = pipeline("text2text-generation", model="google/flan-t5-small")
11
  # checking for ofingpt
12
  # ofintech/FinGPT_0.1.3
13
+ # pipe = pipeline("text2text-generation", model="MudassirFayaz/llama-2-7b_career_0.6.0", trust_remote_code=True)
14
+ # Initialize the text generation pipeline
15
+ # model = AutoModelForSeq2SeqLM.from_pretrained("ofintech/FinGPT_0.1.3")
16
+ pipe = pipeline("text2text-generation", model="MudassirFayaz/llama-2-7b_career_0.6.0", tokenizer=tokenizer)
17
 
18
 
19
  @app.get("/")