GenAIDevTOProd commited on
Commit
58cc85a
·
verified ·
1 Parent(s): 7e48888

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -9,13 +9,8 @@ Original file is located at
9
 
10
  import gradio as gr
11
  import yaml, json, requests
12
- from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
13
  import os
14
 
15
- # Optional fallback: Falcon 1B generator
16
- tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-rw-1b")
17
- model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-rw-1b")
18
- falcon_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
19
 
20
  # Call Mistral-7B via OpenRouter
21
  def call_mistral_openrouter(prompt):
 
9
 
10
  import gradio as gr
11
  import yaml, json, requests
 
12
  import os
13
 
 
 
 
 
14
 
15
  # Call Mistral-7B via OpenRouter
16
  def call_mistral_openrouter(prompt):