SoSa123456 commited on
Commit
e6d9b29
1 Parent(s): f00d034

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -8,7 +8,11 @@ from transformers import pipeline
8
  import gradio as gr
9
 
10
  # Initialize the language model
11
- generator = pipeline('text-generation', model='mistralai/Mixtral-8x7B-Instruct-v0.1') #'tiiuae/falcon-7b-instruct')
 
 
 
 
12
 
13
  def generate_script(host_name, listener_location, causes_climate_change, co2_level, effects_climate_change, sea_level_rise, warming_rate, potential_solutions, individual_role, call_to_action):
14
  # Variables
 
8
  import gradio as gr
9
 
10
  # Initialize the language model
11
+ #generator = pipeline('text-generation', model='mistralai/Mixtral-8x7B-Instruct-v0.1') #'tiiuae/falcon-7b-instruct')
12
+ from huggingface_hub import InferenceClient
13
+ import gradio as gr
14
+
15
+ generator = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
16
 
17
  def generate_script(host_name, listener_location, causes_climate_change, co2_level, effects_climate_change, sea_level_rise, warming_rate, potential_solutions, individual_role, call_to_action):
18
  # Variables