dgnk007 commited on
Commit
35269fa
·
1 Parent(s): 1ed6e54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -1,3 +1,12 @@
1
  import gradio as gr
2
-
3
- gr.Interface.load("models/dgnk007/eagle").launch()
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ import pip
3
+ pip.main(['install', 'transformers'])
4
+ model_name_or_path='dgnk007/eagle'
5
+ def generate_response(message):
6
+ prompt_template=f"Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n {text}\n\n### Response:\n"
7
+ generate=pipeline('text-generation',model=model_name_or_path)
8
+ response=generate(prompt_template,max_length=1024,return_full_text=False)
9
+ return response[0]['generated_text']
10
+
11
+ demo = gr.Interface(fn=generate_response, inputs="image", outputs="label")
12
+ demo.launch(share=True)