Bloodacote commited on
Commit
d690cb1
·
verified ·
1 Parent(s): 2451e00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -1,15 +1,22 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
 
4
- # Initialize the text generation pipeline
 
 
 
 
 
 
 
5
  pipe = pipeline("text-generation", model="meta-llama/CodeLlama-7b-hf")
6
 
7
  def generate_text(prompt):
8
- # Generate text based on the provided prompt using the pipeline
9
  generated_text = pipe(prompt, max_length=50, do_sample=True)[0]['generated_text']
10
  return generated_text
11
 
12
- # Create a Gradio interface
13
  demo = gr.Interface(
14
  fn=generate_text,
15
  inputs="text",
@@ -18,5 +25,5 @@ demo = gr.Interface(
18
  description="Enter a prompt and see the generated text."
19
  )
20
 
21
- # Launch the Gradio interface
22
  demo.launch()
 
1
  import gradio as gr
2
+ from transformers import pipeline, HfApi
3
 
4
+ part1 = hf_rSFLYgnNSgUWyBc
5
+ part2 = BfOwDzWGSnfkeBVbYik
6
+
7
+ # Установка API ключа
8
+ api_token = part1 + part2
9
+ HfApi().token = api_token
10
+
11
+ # Инициализация пайплайна для генерации текста
12
  pipe = pipeline("text-generation", model="meta-llama/CodeLlama-7b-hf")
13
 
14
  def generate_text(prompt):
15
+ # Генерация текста на основе введенного промпта
16
  generated_text = pipe(prompt, max_length=50, do_sample=True)[0]['generated_text']
17
  return generated_text
18
 
19
+ # Создание Gradio интерфейса
20
  demo = gr.Interface(
21
  fn=generate_text,
22
  inputs="text",
 
25
  description="Enter a prompt and see the generated text."
26
  )
27
 
28
+ # Запуск Gradio интерфейса
29
  demo.launch()