Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,7 +9,7 @@ token = os.environ.get("token")
 
 def instantiate_gpt2(model_name: str,max_length_ : int, num_return_sequences : int, text : str) -> dict:
     pipe = pipeline(task='text-generation', model=f'Iscte-Sintra/{model_name}', tokenizer=f'Iscte-Sintra/{model_name}', token=token, truncation=True)
-    if model_name is "Qwen_v0.1":
+    if model_name=="Qwen_v0.1":
         results = pipe(
             text,
             max_new_tokens=max_length_,