diginoron commited on
Commit
d059e89
·
verified ·
1 Parent(s): 1641295

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -1,8 +1,15 @@
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # استفاده از inference از Hugging Face
5
- pipe = pipeline("text-generation", model="google/gemma-1.1-7b-it") # به‌جای 27B چون رایگان نیست
 
 
 
 
 
 
6
 
7
  def generate_topics(field, major, keywords, audience, level):
8
  prompt = f"""
 
1
+ import os
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
+ # گرفتن توکن از متغیر محیطی
6
+ token = os.environ.get("HF_TOKEN")
7
+
8
+ pipe = pipeline(
9
+ "text-generation",
10
+ model="google/gemma-1.1-7b-it",
11
+ token=token
12
+ )
13
 
14
  def generate_topics(field, major, keywords, audience, level):
15
  prompt = f"""