Spaces:

Eminda
/

llm-test

Runtime error

Eminda commited on Jul 7, 2023

Commit

7194457

1 Parent(s): dd8d188

test gpt2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 # Use a pipeline as a high-level helper
 from transformers import pipeline
-pipe = pipeline("text-generation", model="TheBloke/orca_mini_v2_7B-GGML", trust_remote_code=True)
 def greet(name):
     return pipe('who is the president of USA')

 # Use a pipeline as a high-level helper
 from transformers import pipeline
+pipe = pipeline("text-generation", model="gpt2", trust_remote_code=True)
 def greet(name):
     return pipe('who is the president of USA')